Python get_latent_model_configuration 예제들

예제 #1

 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,
     )

예제 #2

파일: seq_seq_domain_translation.py 프로젝트: GVS-Lab/domain_translation

 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())