Update Trainer class to rename parameters for clarity in embedding configuration

2026-01-08 17:11:19 +08:00
parent d20d32ba22
commit d7600514af
1 changed files with 2 additions and 2 deletions
--- a/train.py
+++ b/train.py
@@ -234,8 +234,8 @@ class Trainer:
                n_cont=dataset.n_cont,
                n_cate=dataset.n_cate,
                cate_dims=dataset.cate_dims,
-                pretrained_emd_path=cfg.pretrained_emd_path,
-                freeze_pretrained_emd=True,
+                pretrained_weights_path=cfg.pretrained_emd_path,
+                freeze_embeddings=True,
            ).to(self.device)
        else:
            raise ValueError(f"Unsupported model type: {cfg.model_type}")