taki0112 / StyleGAN-Tensorflow

Simple & Intuitive Tensorflow implementation of StyleGAN (CVPR 2019 Oral)
MIT License
211 stars 60 forks source link

train main #16

Open ucasiggcas opened 4 years ago

ucasiggcas commented 4 years ago

hi, no matter windows or ubuntu gpu, all wrong in gpu

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "main.py", line 102, in <module>
    main()
  File "main.py", line 81, in main
    gan.train()
  File "/home/././StyleGAN-Tensorflow/StyleGAN.py", line 473, in train
    self.d_loss_per_res[current_res]])
  File "/home/./anaconda3/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 956, in run
    run_metadata_ptr)
  File "/home/./anaconda3/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1180, in _run
    feed_dict_tensor, options, run_metadata)
  File "/home/./anaconda3/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1359, in _do_run
    run_metadata)
  File "/home/./anaconda3/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1384, in _do_call
    raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found.
  (0) Invalid argument:  Got 30 frames, but animated gifs can only be decoded by tf.image.decode_gif or tf.image.decode_image
     [[{{node DecodeJpeg}}]]
     [[IteratorGetNext]]
     [[RemoteCall]]
     [[IteratorGetNext_3]]
  (1) Invalid argument:  Got 30 frames, but animated gifs can only be decoded by tf.image.decode_gif or tf.image.decode_image
     [[{{node DecodeJpeg}}]]
     [[IteratorGetNext]]
     [[RemoteCall]]
     [[IteratorGetNext_3]]
     [[gradients_4/AddN_166/_187]]
0 successful operations.
1 derived errors ignored.

and my gpu_num=2 tf-gpu=1.15 could you help me ? thx