Open zhenzi0322 opened 10 months ago
你好,是否可以提供更多的报错信息?
看起来是这一步出问题了: reference_feats = self.memory.read_chars(target_style_ids, trg_sample_index, reduction=reduction)
可以检查一下这一步输出的特征纬度是否正确: self.gen.encode_write_comb(in_style_ids, style_sample_index, in_imgs[0]) # [B*3,256,16,16]
trainer/combined_trainer.py:
上图就是我本地中的输出
看起来是某一次sample的reference个数不足k(num shot)个,但不足k个的batch应该会被丢掉才对,建议你仔细观察得到风格特征过程中tensor的维度变化。
一直没找到原因,我这边就是卡在这一步,训练不了
Were you able to solve the problem? If so, how?
一直没找到原因,我这边就是卡在这一步,训练不了
Were you able to solve the problem? If so, how?
一直没找到原因,我这边就是卡在这一步,训练不了
I was the one who ran successfully before, but I'm not quite sure where the modifications were made
训练时出现如下问题: