Thanks a lot. This is my first topic in the pytorch discuss. I am new to PyTorch.
Thanks.
I also create a new topic in https://discuss.pytorch.org/t/too-many-parameters-cause-single-gpu-memory-out-how-to-solve-it/18973
Could you please have a look?