search

imlixinyang / HiSD

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

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

Open zzk88862 opened 1 year ago

zzk88862 commented 1 year ago

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

image

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

感谢并期待您的回复

imlixinyang commented 11 months ago

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