You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I've noticed that in some parts, the fact that a torch type is float64 is forced. I think this can be problematic when we use Adam with other libraries.
I've noticed that in some parts, the fact that a torch type is float64 is forced. I think this can be problematic when we use Adam with other libraries.
A possibility is to remove the following lines
adam/src/adam/pytorch/torch_like.py
Lines 19 to 22 in a6a60bd
and
adam/src/adam/pytorch/computations.py
Lines 23 to 25 in a6a60bd
so the user can force the float64 with
torch.set_default_dtype(torch.float64)
as is done in the testsThe text was updated successfully, but these errors were encountered: