the training is not complete

[elham1992]

thanks alot for sharing your code.
i use the cli to train on my own dataset .
python main.py --name celeba --train ./data/train --test ./data/test --vgg_path ./data/pytorch_to_keras_vgg16.h5

out put on command prompt is :

why no result and training is 0%??
pleas help me and answer to my Question.

[Deleted user]

e and answer to m

I think your GPU is out of memory.
Make sure not run out of the capability of memory by calling "nvidia-smi"

Best regards,