machine-learning - Stylegan3：如何将特定的刻度状态保存在文件中并从该文件恢复训练？

Question

我正在使用 Google Colab 在我的数据上训练 Stylegan3。我可以从 .pkl 文件恢复。

但我想将特定的刻度状态保存在文件中，然后从该文件恢复训练。我该怎么做？我检查了 PyTorch 模型保存和恢复，但对 Stylegan3 刻度保存和恢复一无所知。

Stylegan3 官方仓库：https ://github.com/NVlabs/stylegan3

Answer 1

解决方案是：

“当您运行 train.py 时，您拥有的选项之一是--snap，即保存训练模型快照的频率（以刻度为单位），因此您可以设置--snap=100.

例如，要从这个刻度恢复，它们将保存为 network-snapshot-000100.pkl，所以当你再次运行 train.py 时，你使用--resume=/path/to/network-snapshot-000100.pkl，在其中输入该 .pkl 文件的实际路径。”

来自https://github.com/NVlabs/stylegan3/issues/67