Open mzweilin opened 1 year ago
We need to allow PyTorch 2.0.0, because PyTorch 2.0.1 gives incomplete dependency to Poetry.
https://github.com/pytorch/pytorch/issues/100974 https://github.com/python-poetry/poetry/issues/7902
An alternative solution is to manually add the missing dependency of PyTorch 2.0.1 to your Poetry config.
I also believe the updated dependency (PyTorch>=2.0.0) makes more sense.
Please check all relevant options.
Please describe the tests that you ran to verify your changes. Consider listing any relevant details of your test configuration.
pytest
CUDA_VISIBLE_DEVICES=0 python -m mart experiment=CIFAR10_CNN_Adv trainer=gpu trainer.precision=16
CUDA_VISIBLE_DEVICES=0,1 python -m mart experiment=CIFAR10_CNN_Adv trainer=ddp trainer.precision=16 trainer.devices=2 model.optimizer.lr=0.2 trainer.max_steps=2925 datamodule.ims_per_batch=256 datamodule.world_size=2
pre-commit run -a
Make sure you had fun coding 🙃
What does this PR do?
We need to allow PyTorch 2.0.0, because PyTorch 2.0.1 gives incomplete dependency to Poetry.
https://github.com/pytorch/pytorch/issues/100974 https://github.com/python-poetry/poetry/issues/7902
An alternative solution is to manually add the missing dependency of PyTorch 2.0.1 to your Poetry config.
I also believe the updated dependency (PyTorch>=2.0.0) makes more sense.
Type of change
Please check all relevant options.
Testing
Please describe the tests that you ran to verify your changes. Consider listing any relevant details of your test configuration.
pytest
CUDA_VISIBLE_DEVICES=0 python -m mart experiment=CIFAR10_CNN_Adv trainer=gpu trainer.precision=16
reports 70% (21 sec/epoch).CUDA_VISIBLE_DEVICES=0,1 python -m mart experiment=CIFAR10_CNN_Adv trainer=ddp trainer.precision=16 trainer.devices=2 model.optimizer.lr=0.2 trainer.max_steps=2925 datamodule.ims_per_batch=256 datamodule.world_size=2
reports 70% (14 sec/epoch).Before submitting
pre-commit run -a
command without errorsDid you have fun?
Make sure you had fun coding 🙃