James Betker 036684893e Add LARS optimizer & support for BYOL idiosyncrasies ... - 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)		2020-12-23 20:33:43 -07:00
..
.idea
data	Implement a few changes to support training BYOL networks	2020-12-23 10:50:23 -07:00
models	Add LARS optimizer & support for BYOL idiosyncrasies	2020-12-23 20:33:43 -07:00
scripts	Implement a few changes to support training BYOL networks	2020-12-23 10:50:23 -07:00
trainer	Add LARS optimizer & support for BYOL idiosyncrasies	2020-12-23 20:33:43 -07:00
utils
multi_modal_train.py
process_video.py
requirements.txt
test_image_patch_classifier.py
test.py
train.py	Implement a few changes to support training BYOL networks	2020-12-23 10:50:23 -07:00