James Betker c203cee31e Allow swapping to torch DDP as needed in code		2020-12-09 15:03:59 -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
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	Allow swapping to torch DDP as needed in code	2020-12-09 15:03:59 -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