Hooman Shayani
Hooman Shayani
It seems m=model.to(device) creates a duplicate pointer to the model that is not needed. Just to make it simpler and clearer. :)
For the case when x_t is given as input there was a typo breaking the code.
Are you going to add in-painting?