model = FaceAging(
session, # TensorFlow session
is_training=FLAGS.is_train, # flag for training or testing mode
save_dir=FLAGS.savedir, # path to save checkpoints, samples, and summary
dataset_name=FLAGS.dataset # name of the dataset in the folder ./data
)
in the main.py, all the print commands dose not work in GPU. Colab can work. do you know why?
model = FaceAging( session, # TensorFlow session is_training=FLAGS.is_train, # flag for training or testing mode save_dir=FLAGS.savedir, # path to save checkpoints, samples, and summary dataset_name=FLAGS.dataset # name of the dataset in the folder ./data )
in the main.py, all the print commands dose not work in GPU. Colab can work. do you know why?