FredZhang7
/

efficientnetv25_rw_s

Image Classification

efficientnetv25

Model card Files Files and versions

FredZhang7 commited on Oct 9, 2023

Commit

2995779

·

1 Parent(s): cdbd71c

update code

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -26,13 +26,13 @@ in terms of top-1 accuracy, efficiency, and robustness on my dataset and [CMAD b
 <br>
-### Load 1000 Class PyTorch Jit Model
 ```python
 from transformers import AutoModel
 model = AutoModel.from_pretrained("FredZhang7/efficientnetv2.5_rw_s", trust_remote_code=True)
 ```
-### Prepare Model for Training
 To change the number of classes, replace the linear classification layer.
 Here's an example of how to convert the architecture into a trainable model.
 ```bash

 <br>
+### Load PyTorch Jit Model with 1000 Classes
 ```python
 from transformers import AutoModel
 model = AutoModel.from_pretrained("FredZhang7/efficientnetv2.5_rw_s", trust_remote_code=True)
 ```
+### Load Model with Custom Classes
 To change the number of classes, replace the linear classification layer.
 Here's an example of how to convert the architecture into a trainable model.
 ```bash