lite_hrnet icon indicating copy to clipboard operation
lite_hrnet copied to clipboard

Published 20 hours ago • verified publisher icon wuwusky

error when train

Open Xenos24R opened this issue 2 years ago • 1 comments

when the mode set to train,will raise this error:"Expected more than 1 value per channel when training, got input size torch.Size([1, 2, 1, 1])"

Xenos24R avatar Apr 22 '22 14:04 Xenos24R

maybe u can set "batch_size>1"~~ and set "drop_last=True"

wuwusky avatar Apr 26 '22 01:04 wuwusky