Update README.md
Browse files
README.md
CHANGED
|
@@ -3,8 +3,6 @@ datasets:
|
|
| 3 |
- huggan/wikiart
|
| 4 |
metrics:
|
| 5 |
- accuracy
|
| 6 |
-
base_model:
|
| 7 |
-
- google/vit-base-patch16-224-in21k
|
| 8 |
---
|
| 9 |

|
| 10 |
|
|
@@ -15,4 +13,4 @@ BATCH_SIZE = 16
|
|
| 15 |
WEIGHT_DECAY = 0.05
|
| 16 |
EPOCHS = 100
|
| 17 |
WARMUP_RATIO = 0.1
|
| 18 |
-
```
|
|
|
|
| 3 |
- huggan/wikiart
|
| 4 |
metrics:
|
| 5 |
- accuracy
|
|
|
|
|
|
|
| 6 |
---
|
| 7 |

|
| 8 |
|
|
|
|
| 13 |
WEIGHT_DECAY = 0.05
|
| 14 |
EPOCHS = 100
|
| 15 |
WARMUP_RATIO = 0.1
|
| 16 |
+
```
|