From eec406d79f16228ea855f1dc94a59c380343a58a Mon Sep 17 00:00:00 2001 From: Jiarui Li Date: Thu, 16 Oct 2025 17:10:01 +0800 Subject: [PATCH] update ignored events. --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 92fa286..808be72 100644 --- a/train.py +++ b/train.py @@ -34,7 +34,7 @@ class TrainConfig: # Loss parameters # 0 = padding, 1 = "no event" - ignored_token_ids = [0, 1] + ignored_token_ids = [0, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12] # Example ignored token IDs # System parameters device = 'cuda' if torch.cuda.is_available() else 'cpu'