Closed dxoigmn closed 1 year ago
Usage example:
python -m mart \
experiment=ArmoryCarlaOverObjDet_TorchvisionFasterRCNN \
fit=false \
+trainer.limit_test_batches=1 \
+model.load_state_dict.losses_and_detections.model=/home/weilinxu/coder/GARD-with-MART/oscar/model_zoo/carla_rgb_weights_eval6.pt
What does this PR do?
The mechanism for wrapping a module in
mart.nn.load_state_dict
makes YAML configuration more verbose and is unnecessary if that logic is moved intoLitModular
. Ideally, however, we would support custom syntax within the YAML configuration for things that do not support weight loading. For example: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.
Before submitting
pre-commit run -a
command without errorsDid you have fun?
Make sure you had fun coding 🙃