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

Revert "Add nano and tiny model presets"

Browse Source 
This reverts commit 6f7b5be405.
This commit is contained in:
Jiarui Li
2026-07-23 16:03:29 +08:00
parent 6f7b5be405
commit 22faee7c51
3 changed files with 29 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
models.py
View File

@@ -15,7 +15,7 @@ from backbones import (
from targets import PAD_IDX
EVENT_TRAJECTORY_ARCHITECTURE = "event_trajectory_shared_v2"
EVENT_TRAJECTORY_ARCHITECTURE = "event_trajectory_shared_v1"
@dataclass(frozen=True)
@@ -33,8 +33,7 @@ class EventTrajectoryModelSize:
MODEL_SIZE_PRESETS = {
"nano": EventTrajectoryModelSize(d_model=120, n_trajectory=6),
"tiny": EventTrajectoryModelSize(d_model=256, n_trajectory=8),
"nano": EventTrajectoryModelSize(d_model=256, n_trajectory=8),
"small": EventTrajectoryModelSize(d_model=512, n_trajectory=8),
"medium": EventTrajectoryModelSize(d_model=768, n_trajectory=12),
"huge": EventTrajectoryModelSize(d_model=1024, n_trajectory=16),