Ejemplo n.º 1
0
 def initialize_dcm_model(self):
     model_config = self.latent_dcm_config["model_config"]
     optimizer_config = self.latent_dcm_config["optimizer_config"]
     loss_config = self.latent_dcm_config["loss_config"]
     self.latent_dcm_config = get_latent_model_configuration(
         model_dict=model_config,
         optimizer_dict=optimizer_config,
         loss_dict=loss_config,
         device=self.device,
     )
 def initialize_dcm_model(self):
     model_config = self.latent_dcm_config["model_config"]
     optimizer_config = self.latent_dcm_config["optimizer_config"]
     loss_config = self.latent_dcm_config["loss_config"]
     self.latent_dcm_config = get_latent_model_configuration(
         model_dict=model_config,
         optimizer_dict=optimizer_config,
         loss_dict=loss_config,
         device=self.device,
     )
     self.initial_dcm_weights = copy.deepcopy(
         self.latent_dcm_config["model"].state_dict())