From 3e52285ab67117fdacd42bbb494f95de6ac0b84a Mon Sep 17 00:00:00 2001
From: Colin <>
Date: Sun, 10 Aug 2025 21:13:37 +0800
Subject: [PATCH] Reset Base config.

---
 wit/train.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/wit/train.py b/wit/train.py
index a1f6989..9538dc1 100644
--- a/wit/train.py
+++ b/wit/train.py
@@ -30,7 +30,7 @@ if __name__ == "__main__":
     conf.dataloader_works = 2
 
     conf.dataset.meaning.start = 10000
-    conf.dataset.meaning.end = 200000
+    conf.dataset.meaning.end = 100000
     conf.dataset.meaning.size = None
     conf.dataset.meaning.min_subitem = 2
     conf.dataset.meaning.max_subitem = 6
@@ -50,7 +50,6 @@ if __name__ == "__main__":
     torch.cuda.manual_seed_all(conf.seed)
 
     model = QWenLMHeadModel(conf.model_config)
-    # model = RWKVLMHeadModel(conf.model_config)
     qwen = LightModule(conf, model)
 
     train_dataloader, val_dataloader = m.InitDataset(conf)