当前位置: 首页> 健康> 美食 > 【深度学习】【模型训练】输出轮数卡住不动【解决方案】

【深度学习】【模型训练】输出轮数卡住不动【解决方案】

时间:2025/8/26 22:12:12来源:https://blog.csdn.net/weixin_43031313/article/details/140667968 浏览次数:0次

一、问题描述

如下图,笔者最近做一个Python深度学习项目时,输出迭代轮数时卡住:
在这里插入图片描述
如上图中所示,设置每10轮输出一次损失,相关参考代码如下:

# run style transfer
max_iter = 200  # 最大迭代次数
show_iter = 10  # 打印间隔
while n_iter[0] <= max_iter:def closure():optimizer.zero_grad()out = extract_layers(loss_layers, opt_img, model=vgg)layer_losses = [weights[a] * loss_fns[a](A, targets[a]) for a, A in enumerate(out)]loss = sum(layer_losses)loss.backward()n_iter[0] += 1# print lossif n_iter[0] % show_iter == (show_iter - 1):# show_iter = 10    9# 9, 19, 29, 39, 49print('Iteration: %d, loss: %f' % (n_iter[0] + 1, loss.data))return losswhile n_iter[0] <= max_iter:def closure():optimizer.zero_grad()out = extract_layers(loss_layers, opt_img, model=vgg)layer_losses = [weights[a] * loss_fns[a](A, targets[a]) for a, A in enumerate(out)]loss = sum(layer_losses)loss.backward()n_iter[0] += 1# print lossif n_iter[0] % show_iter == (show_iter - 1):# show_iter = 10    9# 9, 19, 29, 39, 49print('Iteration: %d, loss: %f' % (n_iter[0] + 1, loss.data))return loss

(代码参考书目:《Pytorch深度学习》,【印度】Vishnu Subramanian 著,王海玲 刘江峰 译,人民邮电出版社,北京,2019年4月)

二、解决方案

原本以为是代码运行过程中出现了问题,其实是Windows Powershell终端自身显示的问题。此时,按一次(注意仅一次Ctrl+C组合键,补全显示即可。(按两次会中止程序)。正确运行截图:

在这里插入图片描述

关键字:【深度学习】【模型训练】输出轮数卡住不动【解决方案】

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

责任编辑: