在用自己的图片数据集训练stylegan2模型时，代码跑了一天突然断开，解决方法！

x秀x

于 2021-05-11 15:14:23 发布

阅读量1.4k

点赞数 2

文章标签：机器学习人工智能神经网络深度学习 github

本文链接：https://blog.csdn.net/m0_52122736/article/details/116655697

版权

1，找到训练一半的pkl文件并放置该文件到stylegan2文件夹下的models里

在这里插入图片描述

2，更改training文件下的training——loop.py(131行)
源代码为 resume_pkl = None
把none改为训练一半的pkl文件夹的地址（注意地址之间的间隔符号用/而不是\，例如c;\user…改为c:/user/…）

在这里插入图片描述

3执行代码

python run_training.py --num-gpus=1 --data-dir=./datasets --config=config-f --dataset=custome_dataset  --mirror-augment=true --metric=none --total-kimg=10000 --result-dir="C:\Users\VRLAB\Desktop\stylegan2\stylegan2\results\00045-stylegan2-custome_dataset-1gpu-config-f"

（注意，data-dir=./datasets，dataset=custome_dataset，result-dir=“C:\Users\VRLAB\Desktop\stylegan2\stylegan2\results\00045-stylegan2-custome_dataset-1gpu-config-f”，这三个按照自己文件存放的位置经行修改）

END