girish00 commited on
Commit
953dad7
·
verified ·
1 Parent(s): e42c1f9

make project runnable and endpoint-ready

Browse files
Files changed (1) hide show
  1. training_config.json +12 -0
training_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "Qwen/Qwen2.5-Coder-0.5B-Instruct",
3
+ "dataset_size": 8000,
4
+ "train_file": "train.json",
5
+ "output_dir": "model",
6
+ "hf_repo_id": "",
7
+ "epochs": 3,
8
+ "batch_size": 2,
9
+ "learning_rate": 0.0001,
10
+ "max_length": 512,
11
+ "use_4bit": true
12
+ }