thanks so much for your reply!
so i got the ‘cuda out of memory’  problem exactly due to my GPU memory is not enough?
(my GPU:1080Ti )
but i trained my model on a 24G GPU ,it got the same problems.
it’s really puzzled me