Training data size

[shiminxuan003]

What is the image size used for training CCN,
Why can't the matrix be multiplied when I input 1024 in training,
and The problem of ' 'Nan or Inf found in input tensor '' when inputting 256 during training

[LeslieZhoa]

Hi @shiminxuan003 ! the training input size is 256. If you want to train the 1024 size, just change the size params and add resize in ccnloader

[shiminxuan003]

你好!
我在config.py中将self.size改成了1024,并在ccnloader中将图片resize到1024还是无法训练
我下载了你预训练的256的550000pt,这个文件是只支持256吗?训练1024图像还能继续使用吗?
并且我使用256图像进行训练的时候出现'NaN or Inf found in input tensor.'的问题是什么原因?

[LeslieZhoa]

Hi @shiminxuan003
1024无法预载256的模型的，训练的前提是stylegan2的模型是可以支持生成真实人的，如果你想在1024上训练，建议你找到或者自己基于ffhq训练一个stylegan2的模型。
至于'NaN or Inf found in input tensor.' 我不确定你的stylegan2预载的模型是否有什么问题，或者你的输入输出，学习率设置上，你可以使用pdb进行断点debug，查看输入输出是否是符合预期的。
你可以使用tensorboard查看中间结果，理论上刚开始训练的时候xg和xt的相似度是很大的。
期待你有一个好的结果！