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