DL-Art-School/codes
James Betker db9e9e28a0 Fix an issue where GPU0 was always being used in non-ddp
Frankly, I don't understand how this has ever worked. WTF.
2020-11-12 15:43:01 -07:00
..
.idea
data stylegan2 in ml art school! 2020-11-12 15:42:05 -07:00
metrics
models Fix an issue where GPU0 was always being used in non-ddp 2020-11-12 15:43:01 -07:00
scripts More work on SSIM/PSNR approximators 2020-11-03 08:09:58 -07:00
switched_conv@cb520afd4d
utils latent work 2020-11-06 20:38:23 -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 work in support of training flow networks in tandem with generators 2020-11-04 18:07:48 -07:00
requirements.txt Fix some distributed training snafus 2020-10-27 15:24:05 -06:00
test.py Include psnr in test.py 2020-10-27 10:25:42 -06:00
train.py Fix an issue where GPU0 was always being used in non-ddp 2020-11-12 15:43:01 -07:00
train2.py Fix an issue where GPU0 was always being used in non-ddp 2020-11-12 15:43:01 -07:00