remove old init pretrained call
@@ -65,12 +65,6 @@ class TrainingJob(Job):
self.optimizer = KgeOptimizer.create(config, self.model)
self.kge_lr_scheduler = KgeLRScheduler(config, self.optimizer)
self.loss = KgeLoss.create(config)
- if self.config.get("train.init_pretrained"):
- self.config.log(
- f"Initializing with embeddings stored in {self.config.get('train.init_pretrained')}"
- )
- packaged_model = load_checkpoint(self.config.get("train.init_pretrained"))
- self.model.init_pretrained(packaged_model)
self.abort_on_nan: bool = config.get("train.abort_on_nan")
self.batch_size: int = config.get("train.batch_size")
self.device: str = self.config.get("job.device")
