search

datawhalechina / torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.
MIT License
374 stars 69 forks source link

fix trainer optimizer bug #26

Closed yinpu closed 2 years ago

yinpu commented 2 years ago

optimizer init should after model.to(device)