Method state_dict

opacus/optimizers/optimizer.py:577–578 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

575	return self.original_optimizer.__repr__()
576
577	def state_dict(self):
578	return self.original_optimizer.state_dict()
579
580	def load_state_dict(self, state_dict) -> None:
581	self.original_optimizer.load_state_dict(state_dict)

save_checkpointMethod · 0.45

fixFunction · 0.45

test_get_compatible_module_inactionMethod · 0.45

test_parameters_matchMethod · 0.45

test_checkpointsMethod · 0.45

setUpMethod · 0.45

test_remove_hooksMethod · 0.45

test_state_dictMethod · 0.45

test_load_state_dictMethod · 0.45

setUpMethod · 0.45

no outgoing calls

test_get_compatible_module_inactionMethod · 0.36

test_parameters_matchMethod · 0.36

test_checkpointsMethod · 0.36

setUpMethod · 0.36

test_remove_hooksMethod · 0.36

test_state_dictMethod · 0.36

test_load_state_dictMethod · 0.36

setUpMethod · 0.36

test_comparison_dpoptimizer_vs_adaclip_accountingMethod · 0.36

setUpMethod · 0.36

test_to_standard_moduleMethod · 0.36

test_remove_hooksMethod · 0.36