hackergeek98 commited on
Commit
bd57c6c
·
verified ·
1 Parent(s): 1ebbc02

Create config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +10 -0
config.yaml ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ project_name: phi4-training
2
+ model: microsoft/phi-4 # Replace with "microsoft/phi-4" if available
3
+ data_path: ./
4
+ task: text-generation # Or your task (e.g., text-classification)
5
+ learning_rate: 2e-5
6
+ batch_size: 2
7
+ epochs: 3
8
+ fp16: true
9
+ push_to_hub: true
10
+ repo_id: "hackergeek98/phi4-trained"