jiarui_li/DeepHealth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Trainer class to rename parameters for clarity in embedding configuration

Browse Source
This commit is contained in:
Jiarui Li
2026-01-08 17:11:19 +08:00
parent d20d32ba22
commit d7600514af
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@@ -234,8 +234,8 @@ class Trainer:
n_cont=dataset.n_cont, n_cont=dataset.n_cont,
n_cate=dataset.n_cate, n_cate=dataset.n_cate,
cate_dims=dataset.cate_dims, cate_dims=dataset.cate_dims,
pretrained_emd_path=cfg.pretrained_emd_path, pretrained_weights_path=cfg.pretrained_emd_path,
freeze_pretrained_emd=True, freeze_embeddings=True,
).to(self.device) ).to(self.device)
else: else:
raise ValueError(f"Unsupported model type: {cfg.model_type}") raise ValueError(f"Unsupported model type: {cfg.model_type}")