Closed dxoigmn closed 1 year ago
I released v0.5.1 that reverts the same PR: https://github.com/IntelLabs/MART/releases/tag/v0.5.1
I would like to merge this PR into main
, because Adversary can perform transforms internally in #203
It's better to keep the datamodule transforms simple and consistent with the Armory input format.
You can merge then :)
This reverts commit de77a9d1363f4bbfc94f5301263323c4ce291956.
What does this PR do?
We don't really have multi-modal training working, so revert this until we do. This makes
main
at least usable with our other work.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 🙃