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	Update other docs with dumb config options	2020-12-18 16:21:28 -07:00
multi_modal_train.py	More adjustments to support distributed training with teco & on multi_modal_train	2020-10-27 20:58:03 -06:00
process_video.py	More refactoring	2020-12-18 09:18:34 -07:00
requirements.txt	Update requirements, add image_patch_classifier tool	2020-12-16 09:42:50 -07:00
test_image_patch_classifier.py	More refactoring	2020-12-18 09:18:34 -07:00
test.py	Fix srflow_latent_space_playground bug	2020-12-22 15:42:38 -07:00
train.py	Implement a few changes to support training BYOL networks	2020-12-23 10:50:23 -07:00