DCGAN练习

调用tf.softmax_cross_entropy_with_logits函数出错解决

# loss = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(output, Y)) 
# 因此修改需要成这样： 
loss = tf.reduce_sum(tf.nn.softmax_cross_entropy_with_logits(logits=output, labels=Y))

报错：absl.flags._exceptions.IllegalFlagValueError: flag --train_size=inf: Expect argument to be a string or int, found <class 'float'>

# 在 main.py中的第22行，把 
flags.DEFINE_integer("train_size", np.inf, "The size of train images [np.inf]") 
# 改成 
flags.DEFINE_float("train_size", np.inf, "The size of train images [np.inf]")

pycharm参数设置为：

python3 main.py --input_height 96 --output_height 48 --dataset faces --is_crop True --is_train True --epoch 10