Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
training_arguments = transformers.TrainingArguments(
|
2 |
per_device_train_batch_size=4,
|
3 |
gradient_accumulation_steps=BATCH_SIZE // MICRO_BATCH_SIZE,
|
@@ -28,4 +29,4 @@ MICRO_BATCH_SIZE = 4
|
|
28 |
GRADIENT_ACCUMULATION_STEPS = BATCH_SIZE // MICRO_BATCH_SIZE
|
29 |
LEARNING_RATE = 3e-4
|
30 |
TRAIN_STEPS = 300
|
31 |
-
|
|
|
1 |
+
```
|
2 |
training_arguments = transformers.TrainingArguments(
|
3 |
per_device_train_batch_size=4,
|
4 |
gradient_accumulation_steps=BATCH_SIZE // MICRO_BATCH_SIZE,
|
|
|
29 |
GRADIENT_ACCUMULATION_STEPS = BATCH_SIZE // MICRO_BATCH_SIZE
|
30 |
LEARNING_RATE = 3e-4
|
31 |
TRAIN_STEPS = 300
|
32 |
+
```
|