yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers
MIT License
29.54k stars 8k forks source link

Update main.py for a small optimization #243

Closed Diouo closed 1 year ago

Diouo commented 1 year ago

It is better not to generate variable "inputs" and "targets" repeatedly