【python模型保存之后,再恢复训练如何做】

模型保存之后,再恢复训练如何操作

实际上,想要中断训练,然后保存模型,之后再加载继续训练,并没有那么简单。我们平常保存模型参数一般都是这样遍历之后,再保存到字典里

    for name, data in global_model.state_dict().items():
        diff[name] = (data)
        c = diff[name]

猜你喜欢

转载自blog.csdn.net/m0_54487794/article/details/125100739