Fix Tune

qile2000 · Jan 6, 2025 · 6c3f0af · 6c3f0af
1 parent 466f0b0
commit 6c3f0af
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/LAMDA_TALENT/model/utils.py b/LAMDA_TALENT/model/utils.py
@@ -658,9 +658,7 @@ def objective(trial):
         # method.fit(train_val_data, info, train=True, config=config)  
         # run with this config
         try:
-            set_seeds(args.seed)
-            method = get_method(args.model_type)(args, info['task_type'] == 'regression')   
-            method.fit(train_val_data, info, train=True, config=config,save_model=False)    
+            method.fit(train_val_data, info, train=True, config=config)    
             return method.trlog['best_res']
         except Exception as e:
             print(e)