Adding in some missing cpu() calls and making sure train model from scratch works properly. I think this was already in @lorenzozanisi's pull request from earlier before being closed. I think it is worth it being opened again if you are having success on your end so we are on the same page
Adding in some missing cpu() calls and making sure train model from scratch works properly. I think this was already in @lorenzozanisi's pull request from earlier before being closed. I think it is worth it being opened again if you are having success on your end so we are on the same page