036684893e
- Added LARS and SGD optimizer variants that support turning off certain features for BN and bias layers - Added a variant of pytorch's resnet model that supports gradient checkpointing. - Modify the trainer infrastructure to support above - Fix bug with BYOL (should have been nonfunctional) |
||
---|---|---|
.. | ||
.idea | ||
data | ||
models | ||
scripts | ||
trainer | ||
utils | ||
multi_modal_train.py | ||
process_video.py | ||
requirements.txt | ||
test_image_patch_classifier.py | ||
test.py | ||
train.py |