Update README.md
Browse files
README.md
CHANGED
|
@@ -38,6 +38,8 @@ model = timm.create_model(
|
|
| 38 |
"convnextv2_tiny.fcmae_ft_in22k_in1k_384",
|
| 39 |
pretrained=False,
|
| 40 |
num_classes=39,
|
|
|
|
|
|
|
| 41 |
)
|
| 42 |
|
| 43 |
# load weights
|
|
|
|
| 38 |
"convnextv2_tiny.fcmae_ft_in22k_in1k_384",
|
| 39 |
pretrained=False,
|
| 40 |
num_classes=39,
|
| 41 |
+
drop_rate=0.2,
|
| 42 |
+
drop_path_rate=0.2,
|
| 43 |
)
|
| 44 |
|
| 45 |
# load weights
|