Esempi in Python per BaseModel.load_state_dict

Esempio n. 1

File: model_checkpoint.py Progetto: hanyeliu/deeppointcloud-benchmarks

 def initialize_model(self, model: BaseModel, weight_name: str = None):
     if not self._checkpoint.is_empty:
         state_dict = self._checkpoint.get_state_dict(weight_name)
         model.load_state_dict(state_dict)
         optimizer = self._checkpoint.get_optimizer()
         lr_params = self._checkpoint.get_lr_params()
         model.set_optimizer(optimizer.__class__, lr_params=lr_params)

Esempio n. 2

File: model_checkpoint.py Progetto: sayakgis/deeppointcloud-benchmarks

 def _initialize_model(self, model: BaseModel, weight_name):
     if not self._checkpoint.is_empty:
         state_dict = self._checkpoint.get_state_dict(weight_name)
         model.load_state_dict(state_dict)
         if self._resume:
             model.optimizer = self._checkpoint.get_optimizer(model)
             model.schedulers = self._checkpoint.get_schedulers(model)