lanna0504

> > > >  把这个默认的参数4改成10就行 > > > > > > > > > 十分感谢！但是为什么我运行一直都报out of memory 的错误， 我用的是250G内存的机器，这个也不够吗？ > > > 另外，请问你可以成功训练了吗？ > > > > > >...

> 抱歉，时间隔得太久了，我不大能记清了。我应该没有跑完吧，我刚才看了一眼这个training script里面写的2000个epoch，我估计当时训练了太久仅仅只有10几个epoch就花了10个小时，我就没训练了吧。我觉得应该不是机器的问题，应该就是code base有些bug。 噢噢好的，谢谢

> raise ValueError(f"Target size ({target.size()}) must be the same as input size ({input.size()})") ValueError: Target size (torch.Size([128, 300, 10])) must be the same as input size (torch.Size([128, 300, 4])) >...