Update README.md
Browse files
README.md
CHANGED
@@ -37,3 +37,12 @@ model = fastvit_ma36(
|
|
37 |
)
|
38 |
model.summary()
|
39 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
)
|
38 |
model.summary()
|
39 |
```
|
40 |
+
|
41 |
+
|
42 |
+
## Weight file suffix clarification:
|
43 |
+
|
44 |
+
|
45 |
+
* `keras_fastvit_ma36.weights.h5` - Weights for model variant - `ma36`
|
46 |
+
* `keras_fastvit_ma36_reparam.weights.h5` - Reparameterized weights for model variant - `ma36` (`inference_mode = True`)
|
47 |
+
* `keras_fastvit_ma36_kd.weights.h5` - Weights for model variant - `ma36`, trained using knowledge distillation.
|
48 |
+
* `keras_fastvit_ma36_reparam_kd.weights.h5` - Reparameterized weights for model variant - `ma36`, trained using knowledge distillation (`inference_mode = True`)
|