FredZhang7 commited on
Commit
6bab5de
·
1 Parent(s): 642ba04

update classification layer info

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -15,7 +15,7 @@ I built EfficientNetV2.5 s to outperform the existing EfficientNet b0 to b4, Eff
15
  - Params: 16.64 M
16
  - Multiply-Add Operations: 5.32 G
17
  - Image size: train = 299x299 / 304x304, test = 304x304
18
- - Classification layer: included, and defaults to 1,000 classes
19
  - **Papers:**
20
  - EfficientNetV2: Smaller Models and Faster Training: https://arxiv.org/abs/2104.00298
21
  - Layer-adaptive sparsity for the Magnitude-based Pruning: https://arxiv.org/abs/2010.07611
 
15
  - Params: 16.64 M
16
  - Multiply-Add Operations: 5.32 G
17
  - Image size: train = 299x299 / 304x304, test = 304x304
18
+ - Classification layer: defaults to 1,000 classes, but is empty
19
  - **Papers:**
20
  - EfficientNetV2: Smaller Models and Faster Training: https://arxiv.org/abs/2104.00298
21
  - Layer-adaptive sparsity for the Magnitude-based Pruning: https://arxiv.org/abs/2010.07611