James Betker 97ff25a086 BYOL! ... Man, is there anything ExtensibleTrainer can't train? :)		2020-12-08 13:07:53 -07:00
..
archs	spinenet latent playground!	2020-12-05 20:30:36 -07:00
byol	BYOL!	2020-12-08 13:07:53 -07:00
eval	Directly use dataset keys	2020-12-04 20:14:53 -07:00
experiments	Add in experiments hook	2020-09-19 10:05:25 -06:00
flownet2@db2b7899ea	Update flownet submodule	2020-10-24 11:59:00 -06:00
steps	Fix DDP errors for discriminator	2020-12-07 12:50:57 -07:00
__init__.py	Lots of new discriminator nets	2020-11-10 16:06:54 -07:00
base_model.py	Go back to apex DDP, fix distributed bugs	2020-12-04 16:39:21 -07:00
ExtensibleTrainer.py	Directly use dataset keys	2020-12-04 20:14:53 -07:00
feature_model.py	Directly use dataset keys	2020-12-04 20:14:53 -07:00
loss.py	More adjustments to support multi-modal training	2020-10-22 16:49:34 -06:00
lr_scheduler.py
networks.py	BYOL!	2020-12-08 13:07:53 -07:00