Method state_dict

classification/utils.py:380–381 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

378	return norm
379
380	def state_dict(self):
381	return self._scaler.state_dict()
382
383	def load_state_dict(self, state_dict):
384	self._scaler.load_state_dict(state_dict)

extract_mlp.pyFile · 0.80

load_compress_modelFunction · 0.80

load_pretrainedFunction · 0.80

save_checkpointFunction · 0.80

trainFunction · 0.80

save_checkpointFunction · 0.80

make_deltaFunction · 0.80

apply_deltaFunction · 0.80

safe_save_model_for_hf_trainerFunction · 0.80

no outgoing calls

no test coverage detected