Hugging Face's logo

Spaces:
Tonic
/
SmolFactory
Running

App Files Community
Tonic commited on
Commit
b4f1cb3
·
1 Parent(s): c68717e

adds parameters to medical config

Browse files
Files changed (1) hide show
  1. config/train_gpt_oss_medical_o1_sft.py +1 -1
config/train_gpt_oss_medical_o1_sft.py CHANGED
@@ -55,7 +55,7 @@ config = GPTOSSEnhancedCustomConfig(
55
  # ============================================================================
56
  # TRAINING HYPERPARAMETERS
57
  # ============================================================================
58
- num_train_epochs=1.0,
59
  batch_size=4,
60
  gradient_accumulation_steps=4,
61
  learning_rate=2e-4,
 
55
  # ============================================================================
56
  # TRAINING HYPERPARAMETERS
57
  # ============================================================================
58
+ num_train_epochs=2.0,
59
  batch_size=4,
60
  gradient_accumulation_steps=4,
61
  learning_rate=2e-4,