About the training

[srymaker]

Thanks for your great work!
When I change timesteps from 25 to 15 for training, an error is reported that “RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn”