imlixinyang / HiSD

Official pytorch implementation of paper "Image-to-image Translation via Hierarchical Style Disentanglement" (CVPR 2021 Oral).

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

训练256x256的模型

NROwind opened this issue · comments

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

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

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