Update README.md
Browse files
README.md
CHANGED
@@ -53,10 +53,10 @@ system to improve its camera capabilities, as well as improve the quality of its
|
|
53 |
|
54 |
### Training Parameters
|
55 |
```python
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
```
|
|
|
53 |
|
54 |
### Training Parameters
|
55 |
```python
|
56 |
+
num_train_epochs=1,
|
57 |
+
warmup_steps = 500,
|
58 |
+
per_device_train_batch_size=4,
|
59 |
+
per_device_eval_batch_size=4,
|
60 |
+
weight_decay = 0.01,
|
61 |
+
gradient_accumulation_steps=16
|
62 |
```
|