Step3 出错

[yayinliqing]

当我运行python train.py --opt options/train/NYUv2_BSR/train_stage3.yml时报错

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, 144, 1, 1])

[ByChelsea]

Could you provide more detailed error information? The information provided is too limited for me to determine where the problem lies.