Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -102,7 +102,8 @@ print ("lm datasets")
|
|
| 102 |
####################################################
|
| 103 |
|
| 104 |
#Training Args
|
| 105 |
-
|
|
|
|
| 106 |
training_args = TrainingArguments(
|
| 107 |
output_dir="./",
|
| 108 |
overwrite_output_dir = 'True',
|
|
|
|
| 102 |
####################################################
|
| 103 |
|
| 104 |
#Training Args
|
| 105 |
+
batch_size = 2
|
| 106 |
+
|
| 107 |
training_args = TrainingArguments(
|
| 108 |
output_dir="./",
|
| 109 |
overwrite_output_dir = 'True',
|