繁荣大神

文章
8
资源
0
加入时间
2年10月17天

使用tensorflow训练数据时遇到的问题总结

1、OP_REQUIRES failed at assign_op.h models这个问题的根源在于GPU不够用,解决方法也是各不相同,这里写一下几个共性方法。在eval文件中加入os.environ['CUDA_VISIBLE_DEVICES']='2'强制使用CPU验证降低batch_size修改tensorflow-gpu的版本,可能有效果更换网络2、Argument must be a dense tensor: range(0, 3) - got shape [3], bu