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

一、问题描述

如下图，笔者最近做一个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 =