Update README.md
Browse files
README.md
CHANGED
|
@@ -31,6 +31,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
|
| 31 |
```
|
| 32 |
# Training Details
|
| 33 |
- Trained for three epochs
|
|
|
|
| 34 |
- Learning rate: 5e-5
|
| 35 |
- Optimizer: AdamW
|
| 36 |
- Batch size: 64
|
|
|
|
| 31 |
```
|
| 32 |
# Training Details
|
| 33 |
- Trained for three epochs
|
| 34 |
+
- Resolution: 256x256
|
| 35 |
- Learning rate: 5e-5
|
| 36 |
- Optimizer: AdamW
|
| 37 |
- Batch size: 64
|