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

关于更高分辨率模型的训练问题

zzk88862 opened this issue · comments

commented

你好,非常感谢您的研究并开源如此优秀的项目,
现在,我想基于自己的数据,训练一个高分辨率的生成模型,比如512,我应该如何去调整配置文件中各个通道的维度和尺寸呢,如下是256*256生成器的设置,

image

如果我想训练512*512的生成器,这些参数如何调整吗

感谢并期待您的回复

你好,抱歉回复稍慢了,你可以参考一下configs中128和256的区别(主要在于一些image_size的改变以及相应网络的block数目),当然也会取决于你的机器配置。