使用DCGAN-tensorflow-master出现错误

本文记录了一次使用DCGAN-tensorflow训练MNIST数据集时遇到的问题及解决方案。在运行过程中出现TypeError异常,涉及'tensorflow'库中'sigmoid_cross_entropy_with_logits()'函数的参数错误。通过将参数'labels'替换为'targets',成功解决了该问题。
部署运行你感兴趣的模型镜像

转自:http://bbs.youkuaiyun.com/topics/392087841


我的电脑是windows7系统,安装了anaconda,里面是python3.5。

按照DCGAN-tensorflow-master的readme

在命令行中,转换到工作目录下,
第一步输入了python download.py mnist

第二步输入main.py --dataset mnist --input_height=28 --output_height=28 --c_dim=1 --is_train

这个时候报错了,最后几行显示是:
File "....\DCGAN-tensorflow-master\model.py", line 119, in build_model
logits=self.D_logits, labels=tf.ones_like(self.D)))
TypeError: sigmoid_cross_entropy_with_logits() got an unexpected keyword argument 'labels'
请问有人碰到过吗

----------------------------

change labels  to targets, It is enough.

修改119行以及下面相邻的两个(总共修改三个就可以)

您可能感兴趣的与本文相关的镜像

TensorFlow-v2.15

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值