search

Z-Zheng / SimpleCV

Simplify training, evaluation, prediction in Pytorch
MIT License
63 stars 12 forks source link

fixed a bug in optimizer #7

Closed Z-Zheng closed 5 years ago

Z-Zheng commented 5 years ago

pass all parameters of model into optimizer rather than only trainable part for universal checkpoint operations