Open nghi-truyen opened 4 months ago
We can add a method to save the gradients when optimizing the model by gradient-based optimization methods. New variable "iter_dist_grad" would be added into return_options.
We can add a method to save the gradients when optimizing the model by gradient-based optimization methods. New variable "iter_dist_grad" would be added into return_options.