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
fix(ignite): making state saveable, param_groups modifieable
It was impossible to save the state because of:
* mlverse#1233
* undefined tensors were part of the state such as
max_exp_avg_sq for adam with amsgrad = FALSE.
We now keep them as 0-sized tensors as undefined tensors
are not serializeable.
(The reason we keep them at all is that it simplifies the
saving and loading of state dicts easier)
This PR also improves the tests by removing an unnecessary
call to `torch_manual_seed()` that made the tests deterministic
Created on 2025-01-15 with reprex v2.1.1
The text was updated successfully, but these errors were encountered: