Using the variable "model" after declaring variable "m"
klivin opened this issue · comments
Kevin Weiler commented
When you call m = model.to(device)
, it returns a model that shares the same parameters with the original model but is located on the specified device.
So, in your training loop and anywhere else you use the model after this point, you should use m, not model.
Kevin Weiler commented
duplicate of #18