thograce / C2FNet

Context-aware Cross-level Fusion Network for Camouflaged Object Detection
48 stars 18 forks source link

MyTrain.py #9

Closed leezanla closed 3 months ago

leezanla commented 3 months ago

Hello, may I ask why the following error occurs when I run MyTrain.py file?

raise ValueError(f"Expected more than 1 value per channel when training, got input size {size}")

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