search

hujie-frank / SENet

Squeeze-and-Excitation Networks
Apache License 2.0
3.38k stars 837 forks source link

Enough memory for training but not enough memory for testing? #50

Closed SingL3 closed 6 years ago

SingL3 commented 6 years ago

Greeting! I am training SE-Resnet-50, and I have enough GPU memory for training. But when I am trying to do testing with the same config, I got out of memory error. I am confused, do SE need more memory in testing phase than training?

hujie-frank commented 6 years ago

No, SE doesn't need more memory in testing phase. Maybe the batch size in testing is set too large, or the memory is not thoroughly released after training stage.