Uni-Core icon indicating copy to clipboard operation
Uni-Core copied to clipboard

Assert batch_size==1 error when batch_size is higher than 1

Open dingquanyu opened this issue 2 years ago • 1 comments
trafficstars

Hi,

Thanks for developing this powerful package. When I trained a programme using a batch_size higher than 1, Uni-Core will assert whether the batch_seize is equal to 1 and return AssertionError. Is this normal or it will affect downstream processes?

dingquanyu avatar Feb 03 '23 12:02 dingquanyu

can you provide the settings? If you use the per-sample gradient clipping, the batch size is limited to 1.

guolinke avatar Feb 25 '23 07:02 guolinke