Closed mzweilin closed 2 months ago
Why is this a lightning Callback?
Anomalib performs type checking when instantiate our Callback, so I have to make it a Callback explicitly. We also have the generic instantiate function/class that can instantiate any object.
What does this PR do?
This PR add utils to instantiate from config files generated by
python -m mart.generate_config
.(It should be better if we can get rid of the intermediate config files in the future.)
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 🙃