ISdept commited on
Commit
d7bab5f
·
verified ·
1 Parent(s): 9364d1b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -64,5 +64,6 @@
64
  "optimizer_eps": 1e-08,
65
  "optimizer_weight_decay": 1e-06,
66
  "scheduler_name": "cosine",
67
- "scheduler_warmup_steps": 500
 
68
  }
 
64
  "optimizer_eps": 1e-08,
65
  "optimizer_weight_decay": 1e-06,
66
  "scheduler_name": "cosine",
67
+ "scheduler_warmup_steps": 500,
68
+ "num_inference_steps": 100
69
  }