Searched refs:SpectralNormLoadStateDictPreHook (Results 1 – 1 of 1) sorted by relevance
183 module._register_load_state_dict_pre_hook(SpectralNormLoadStateDictPreHook(fn))189 class SpectralNormLoadStateDictPreHook: class362 if isinstance(hook, SpectralNormLoadStateDictPreHook) and hook.fn.name == name: