Closed peggyxpxu closed 8 months ago
Reduce here instead of bs:
https://github.com/TaoRuijie/Loss-Gated-Learning/blob/main/Stage2/dataLoader.py#L119
Reduce here instead of bs:
https://github.com/TaoRuijie/Loss-Gated-Learning/blob/main/Stage2/dataLoader.py#L119
tks, but I find another error when training Stage2:
Traceback (most recent call last):
File "main_train.py", line 59, in
Sorry I am not sure why you have the number '5130' here...also sorry for missing your information
When I train Stage2, GPU is 11G, batchsize is 1. But it keeps showing 'CUDA out of memory'