train出现尺寸不对

[zzc-ai]

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1])

[bubbliiiing]

你是不是在B站问过啦

[zzc-ai]

对，已经解决啦，多谢大佬

[BugOOM]

同问

[guanbin1994]

对，已经解决啦，多谢大佬

怎么解救呀

[bubbliiiing]

batch设置偶数看看把

[ZoeM2016]

大佬，我把batchsize已经设置为2，但是在多卡训练的时候还是出现了这个错误，该怎么办呢 raise ValueError('Expected more than 1 value per channel when training, got input size {}'.format(size))
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1])
Init_Epoch = 100
Freeze_Epoch = 50
Freeze_batch_size = 2
UnFreeze_Epoch = 150
Unfreeze_batch_size = 2

[bubbliiiing]

你几个卡？

[praymich]

@bubbliiiing 大佬，我也遇到这个问题了，batch设成偶数还是会报错，有四张卡

[bubbliiiing]

分配到每个卡上的是偶数就可以应该

[shijiechua]

@bubbliiiing 大佬，代码里面batch已经是偶数了，但在服务器上只能跑50轮，第51轮就报这个错误

[bubbliiiing]

截图看看呢

[bubbliiiing]

你的训练参数

[shijiechua]

[bubbliiiing]

……你服务器单卡？