为什么到第963 个batch训练结束就不进行下去了,怎么解决这个问题
收起
140行的module去掉试下,直接 model.state_dict()
这样的话第963个batch结束后,紧接着又从第0个开始了
那现在还报错吗
不报不报错了,图片我发提问里面了,这样正常吗
报告相同问题?
程序员都在用的中文IT技术交流社区
专业的中文 IT 技术社区,与千万技术人共成长
关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!