James Betker 0eb1f4dd67 Revert "Get rid of CUDA_VISIBLE_DEVICES" ... It is actually necessary for training in distributed mode. Only do it then.		2020-12-31 10:31:40 -07:00
..
.idea
data	Disable refs and centers altogether in single_image_dataset	2020-12-31 10:13:24 -07:00
models	Add gp debug (fix)	2020-12-30 15:26:54 -07:00
scripts	byol_resnet_playground	2020-12-31 10:15:04 -07:00
trainer	Add sr_fid evaluator	2020-12-30 20:18:58 -07:00
utils	Get rid of CUDA_VISIBLE_DEVICES	2020-12-31 10:13:58 -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	Revert "Get rid of CUDA_VISIBLE_DEVICES"	2020-12-31 10:31:40 -07:00