Commit ·
efa4892
1
Parent(s): 19fce4d
add heading
Browse files
README.md
CHANGED
|
@@ -21,6 +21,7 @@ To be clear, this model is tailored to my image and video classification tasks,
|
|
| 21 |
- **Pretrained:** Yes, but requires finetuning
|
| 22 |
- **Original:** This model architecture is original
|
| 23 |
|
|
|
|
| 24 |
To change the number of classes, replace the linear classification layer.
|
| 25 |
Here's an example to convert the architecture into a training-ready model.
|
| 26 |
```bash
|
|
|
|
| 21 |
- **Pretrained:** Yes, but requires finetuning
|
| 22 |
- **Original:** This model architecture is original
|
| 23 |
|
| 24 |
+
### Prepare Model for Training
|
| 25 |
To change the number of classes, replace the linear classification layer.
|
| 26 |
Here's an example to convert the architecture into a training-ready model.
|
| 27 |
```bash
|