Update README.md
Browse files
README.md
CHANGED
|
@@ -320,25 +320,18 @@ resnet = CustomResNetModel.from_pretrained(
|
|
| 320 |
"final-project-5190/model-resnet-50-base",
|
| 321 |
model_name="microsoft/resnet-50"
|
| 322 |
)
|
| 323 |
-
```
|
| 324 |
-
|
| 325 |
|
| 326 |
-
```python
|
| 327 |
#convnext
|
| 328 |
convnext=CustomConvNeXtModel.from_pretrained(
|
| 329 |
"final-project-5190/model-convnext-tiny-reducePlateau",
|
| 330 |
model_name="facebook/convnext-tiny-224")
|
| 331 |
-
```
|
| 332 |
|
| 333 |
-
```python
|
| 334 |
#vit
|
| 335 |
vit = CustomViTModel.from_pretrained(
|
| 336 |
"final-project-5190/model-ViT-base",
|
| 337 |
model_name="google/vit-base-patch16-224"
|
| 338 |
)
|
| 339 |
-
```
|
| 340 |
|
| 341 |
-
```python
|
| 342 |
#efficientnet
|
| 343 |
efficientnet = CustomEfficientNetModel.from_pretrained(
|
| 344 |
"final-project-5190/model-efficientnet-b0-base",
|
|
|
|
| 320 |
"final-project-5190/model-resnet-50-base",
|
| 321 |
model_name="microsoft/resnet-50"
|
| 322 |
)
|
|
|
|
|
|
|
| 323 |
|
|
|
|
| 324 |
#convnext
|
| 325 |
convnext=CustomConvNeXtModel.from_pretrained(
|
| 326 |
"final-project-5190/model-convnext-tiny-reducePlateau",
|
| 327 |
model_name="facebook/convnext-tiny-224")
|
|
|
|
| 328 |
|
|
|
|
| 329 |
#vit
|
| 330 |
vit = CustomViTModel.from_pretrained(
|
| 331 |
"final-project-5190/model-ViT-base",
|
| 332 |
model_name="google/vit-base-patch16-224"
|
| 333 |
)
|
|
|
|
| 334 |
|
|
|
|
| 335 |
#efficientnet
|
| 336 |
efficientnet = CustomEfficientNetModel.from_pretrained(
|
| 337 |
"final-project-5190/model-efficientnet-b0-base",
|