search

XiangGao1102 / CTSS

MIT License
13 stars 2 forks source link

测试阶段 #4

Open Meimeiainaonao opened 11 months ago

Meimeiainaonao commented 11 months ago

你好，我想你问一下每个数据集中都有不同的预训练模型，我按照python main.py --phase test指令测试，提示我错误如下：checkpoint\AnimeStyle_TWR_g300.0_d300.0_con1.5_color15.0_tv1.0\AnimeStyle.model-70，请问是需要加载.meta还是加载.data-00000-of-00001 屏幕截图 2023-10-12 172430

XiangGao1102 commented 11 months ago

您好，测试时TWR指定epoch=70，CSC和DB指定epoch=80，即可，无须指定.index文件或.meta文件，但二者都是模型加载所需要的文件，只不过tf会自己加载。

OrangeZz0331 commented 10 months ago

我也是这个问题，怎么解决呢？ image

XiangGao1102 commented 10 months ago

你好，预训练的模型，对于TWR数据集用的是第70个epoch的结果，对于另外两个数据集用的是第80个epoch的结果，代码里需要制定epoch数。从报错看，加载的TWR预训练模型，指定了错误的epoch数。 在这里有体现。

在 2023-10-25 17:01:42，"OrangeZz0331" @.***> 写道：

我也是这个问题，怎么解决呢？

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>