tharun2909 commited on
Commit
440b2c8
·
verified ·
1 Parent(s): 9d0d66c

Upload training configuration

Browse files
Files changed (1) hide show
  1. run_config.json +1 -1
run_config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "model_name": "distilgpt2",
3
  "epochs": 2,
4
  "batch_size": 1,
5
  "learning_rate": 0.0003,
 
1
  {
2
+ "model_name": "distilbert/distilgpt2",
3
  "epochs": 2,
4
  "batch_size": 1,
5
  "learning_rate": 0.0003,