Hi Juan,
Sorry for my very late reply and Thanks for your help again.
The data size problem is solved and it is found that the model is not fit for torch summary to calculate the parameter size correctly, and after that my Jetson Nano is gone and I have changed to using Desktop to continuous on the journey, but actually, it is better than using jetson nano as it got more RAM to run instead of 4GB limitation. 
Furthermore, I have got a little breakthrough on the model as well, if you are interested in, here is the notebook link for your review
https://www.kaggle.com/tik65536/low-cost-diamond-feature-preliminary-report 6
I value your comments . 
Thanks,
Dick