train出现尺寸不对 about deeplabv3-plus-pytorch HOT 14 OPEN

你是不是在B站问过啦

from deeplabv3-plus-pytorch.

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

from deeplabv3-plus-pytorch.

同问

from deeplabv3-plus-pytorch.

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

怎么解救呀

from deeplabv3-plus-pytorch.

batch设置偶数看看把

from deeplabv3-plus-pytorch.

大佬，我把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

from deeplabv3-plus-pytorch.

你几个卡？

from deeplabv3-plus-pytorch.

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

from deeplabv3-plus-pytorch.

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

from deeplabv3-plus-pytorch.

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

from deeplabv3-plus-pytorch.

截图看看呢

from deeplabv3-plus-pytorch.

你的训练参数

from deeplabv3-plus-pytorch.

from deeplabv3-plus-pytorch.

……你服务器单卡？

from deeplabv3-plus-pytorch.