RYDE-WORK/lnp_ml
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/lnp_ml.git synced 2026-03-21 09:36:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
RYDE-WORK 2026-01-23 17:51:08 +08:00
parent a56637c8ac
commit 93a6f8654d
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
lnp_ml/modeling/trainer.py
View File

@ -13,18 +13,18 @@ from tqdm import tqdm
@dataclass @dataclass
class LossWeights: class LossWeights:
"""各任务的损失权重""" """各任务的损失权重"""
# size: float = 1.0 size: float = 1.0
# pdi: float = 1.0 pdi: float = 1.0
# ee: float = 1.0 ee: float = 1.0
# delivery: float = 1.0
# biodist: float = 1.0
# toxic: float = 1.0
size: float = 0.1
pdi: float = 0.3
ee: float = 0.3
delivery: float = 1.0 delivery: float = 1.0
biodist: float = 1.0 biodist: float = 1.0
toxic: float = 0.05 toxic: float = 1.0
# size: float = 0.1
# pdi: float = 0.3
# ee: float = 0.3
# delivery: float = 1.0
# biodist: float = 1.0
# toxic: float = 0.05
def compute_multitask_loss( def compute_multitask_loss(