torch.optim.Optimizer.register\_state\_dict\_post\_hook ======================================================= .. currentmodule:: torch.optim .. automethod:: Optimizer.register_state_dict_post_hook