更新时间:2023-12-02 14:17:40
您可以创建一个包含权重和体系结构的tar存档,以及一个包含 model.optimizer.get_state返回的优化器状态的pickle文件.()
.
You could create a tar archive containing the weights and the architecture, as well as a pickle file containing the optimizer state returned by model.optimizer.get_state()
.