Add LogNormalBasisHazardLoss implementation and update training configuration

2026-01-13 15:59:20 +08:00
parent d8b322cbee
commit 1df02d85d7
5 changed files with 431 additions and 15 deletions
--- a/model.py
+++ b/model.py
@@ -2,7 +2,6 @@ import numpy as np
 from typing import Optional, List
 from backbones import Block
 from age_encoder import AgeSinusoidalEncoder, AgeMLPEncoder
-import torch.nn.functional as F
 import torch.nn as nn
 import torch