Closed atop0914 closed 5 years ago
输出信息贴一下,谢谢合作。你如果是用CPU训练的话,确实可能会出现较长时间的无输出情况。因为代码里是1000个epoch输出一次,CPU训练慢,所以你感觉没输出。
if epoch % 1000 == 0:
print('Epoch: %d, batch: %d, training loss: %.6f' % (epoch, batch, loss))
你可以试着把把epoch % 1000
改为epoch % 10
,再看下是否有卡住的情况。
第一轮训练就一直卡在那边