Python DCGAN.add_optimizer示例

示例#1

文件： train.py 项目： LZ-QWQ/GANs_anime_face

from model import DCGAN
import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

with tf.Session() as sess:
    dcgan = DCGAN(gpu_nums=1)
    dcgan.build_model_and_loss()
    dcgan.add_optimizer()
    dcgan.train(sess, epochs=250, batch_size=128, data_dir='data')