Upload training configuration
Browse files- 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,
|