首先参考https://github.com/PeterWang512/GANSketching 下载数据集并设置路径 运行pretrain.py对模型进行预训练,其中pretrain.py使用计图完成,pretrain_torch.py使用pytorch完成
运行train.py对模型进行训练
| Name | Name | Last commit date | ||
|---|---|---|---|---|
首先参考https://github.com/PeterWang512/GANSketching 下载数据集并设置路径 运行pretrain.py对模型进行预训练,其中pretrain.py使用计图完成,pretrain_torch.py使用pytorch完成
运行train.py对模型进行训练