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',
|