NVIDIA / vid2vid

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.
Other
8.6k stars 1.2k forks source link

Training fails when "Updating training sequence length to 4", and empty web/images folder #72

Open yuanzhou15 opened 6 years ago

yuanzhou15 commented 6 years ago

!python train.py --name wtrain --loadSize 512 --n_frames_total 2 --max_frames_per_gpu 2 --dataroot datasets/Weather2 --n_gpus_gen -1 I feel like I'm missing something obvious but I can't figure out how to get past this. My images are 256x182

The error message:

--------- Updating training sequence length to 4 --------- Traceback (most recent call last): File "train.py", line 329, in train() File "train.py", line 96, in train for i in range(0, n_frames_total-t_len+1, n_frames_load): ValueError: range() arg 3 must not be zero

Any help would be much appreciated!

Update: when i run the following : !python train.py --name wtrain --loadSize 256 --n_frames_total 1 --niter_step 10 --dataroot datasets/Weather2 --n_gpus_gen -1

The training ends like this:

End of epoch 19 / 20 Time Taken: 0 sec saving the model at the end of epoch 19, iters 38 update learning rate: 0.000040 -> 0.000020 update learning rate: 0.000040 -> 0.000020 End of epoch 20 / 20 Time Taken: 0 sec saving the model at the end of epoch 20, iters 40 update learning rate: 0.000020 -> 0.000000 update learning rate: 0.000020 -> 0.000000 --------- Updating training sequence length to 4 ---------

When I go to the image folder under ./checkpoints/wtrain/web/images it's empty and I can't see the results of training. Is that where I should be looking for the results?