shirgur / hp-vae-gan Goto Github PK

Hi, first of all thanks for your nice project and paper.

When i try to train with some image, there is issue that GPU is lost.
The version of pytorch is 1.6.0
and i train with GTX 1080 with cuda 11.2 and TITAN RTX with cuda 11.4.

Sometimes train is finished with no error, but sometimes GPU is lost and train is stop.

Were there any issues like this?

Hi, first of all thanks for sharing your code :)

I'm running COLAB notebook with a short video (2 secs). When I run

!python /content/hp-vae-gan/train_video_baselines.py --video-path /content/movie-2secs.mp4 --checkname mymovie--visualize --generator GeneratorSG --train-depth 1

I always get this error in step 10, no matter the number of iterations (I've lowered to 50 in this example):

RuntimeError: CUDA out of memory. Tried to allocate 270.00 MiB (GPU 0; 15.90 GiB total capacity; 14.64 GiB already allocated; 49.88 MiB free; 14.94 GiB reserved in total by PyTorch)
Training scale [10/10]: 0% 0/50

Is this an error? Can I tweak something to make this work? Many thanks in advance

can you please add a google colab

i like this project. tell me, what command can i use for test paint-to-image?

Thank you for opening the source of the excellent work! I noticed that this work first proposed the SVFID metric, however, this seems not included in the code. Could you kindly sharing the code for evaluating the SVFID?