YadongLau

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

5

When I train my coco-style dataset by use : bash train_coco.sh then the errors as follows: Namespace(backbone='resnet', base_size=513, batch_size=4, checkname='deeplab-resnet', crop_size=513, cuda=True, dataset='coco', epochs=10, eval_interval=1, freeze_bn=False, ft=False, gpu_ids=[0], loss_type='ce', lr=0.01,...