search

victoresque / pytorch-template

PyTorch deep learning projects made easy.
MIT License
4.7k stars 1.08k forks source link

Fix optimizer initialization order #83

Closed SunQpark closed 3 years ago

SunQpark commented 3 years ago

Fix issue #82, which points out that optimizer initialization should be done after model parameters are moved to GPU.