Python create_model示例

示例#1

 def _create_model(self):
     config = self.config
     self.model = create_model(config)
     self.model = self.model.to(get_default_device())
     self.optimizer = optim.Adam(self.model.parameters(),
                                 lr=self.config.train.learning_rate,
                                 weight_decay=self.config.train.l2_reg)
     self.gradient_clip = config.train.gradient_clip

示例#2

 def _create_model(self):
     config = self.config
     self.model = create_model(config)
     self.model = try_gpu(self.model)

示例#3

文件： eval_run.py 项目： sonoisa/phrasenode

 def _create_model(self):
     config = self.config
     self.model = create_model(config)
     self.model = self.model.to(get_default_device())