search

imlixinyang / HiSD

Code for "Image-to-image Translation via Hierarchical Style Disentanglement" (CVPR 2021 Oral).
Other
392 stars 47 forks source link

训练256x256的模型 #16

Closed NROwind closed 3 years ago

NROwind commented 3 years ago

你好，我看发布的代码里有train.py，给的训练示例是用的128（即celeba-hq.yaml）的那个配置，如果我想用256的那个celeba-hq_256.yaml来train，我应该在代码中做怎样的修改呢，我试着直接更改使用的yaml文件但是行不通

imlixinyang commented 3 years ago

需要使用其他的config文件 只需要更改你的config文件后 在训练时的--config 指定你需要使用的config文件 核心代码中不需要进行更改 如果有报错信息请把报错发到issue里哦

NROwind commented 3 years ago

谢谢您的答复，是可以训的，不过在我的机器上训256的时间比较长，之前误以为训不了