Skip to content

Commit 51e0a67

Browse files
qubvelqubvel
committed
Fixing table
1 parent 06199b0 commit 51e0a67

File tree

1 file changed

+80
-81
lines changed

1 file changed

+80
-81
lines changed

‎docs/encoders.rst

Lines changed: 80 additions & 81 deletions
Original file line numberDiff line numberDiff line change
@@ -57,84 +57,83 @@ For each encoder, the table below provides detailed information:
5757
Indicates whether the encoder can be exported using ``torch.export.export``, making it suitable for deployment in different environments (e.g., ONNX).
5858

5959

60-
+----------------------------+--------------------------------------+-----------+--------+---------+--------+
61-
| Encoder | Pretrained Weights | Params, M | Script | Compile | Export |
62-
+============================+======================================+===========+========+=========+========+
63-
| resnet18 | imagenet<br>ssl<br>swsl | 11M | ✅ | ✅ | ✅ |
64-
| resnet34 | imagenet | 21M | ✅ | ✅ | ✅ |
65-
| resnet50 | imagenet<br>ssl<br>swsl | 23M | ✅ | ✅ | ✅ |
66-
| resnet101 | imagenet | 42M | ✅ | ✅ | ✅ |
67-
| resnet152 | imagenet | 58M | ✅ | ✅ | ✅ |
68-
| resnext50_32x4d | imagenet<br>ssl<br>swsl | 22M | ✅ | ✅ | ✅ |
69-
| resnext101_32x4d | ssl<br>swsl | 42M | ✅ | ✅ | ✅ |
70-
| resnext101_32x8d | imagenet<br>instagram<br>ssl<br>swsl | 86M | ✅ | ✅ | ✅ |
71-
| resnext101_32x16d | instagram<br>ssl<br>swsl | 191M | ✅ | ✅ | ✅ |
72-
| resnext101_32x32d | instagram | 466M | ✅ | ✅ | ✅ |
73-
| resnext101_32x48d | instagram | 826M | ✅ | ✅ | ✅ |
74-
| dpn68 | imagenet | 11M | ❌ | ✅ | ❌ |
75-
| dpn68b | imagenet+5k | 11M | ❌ | ✅ | ❌ |
76-
| dpn92 | imagenet+5k | 34M | ❌ | ✅ | ❌ |
77-
| dpn98 | imagenet | 58M | ❌ | ✅ | ❌ |
78-
| dpn107 | imagenet+5k | 84M | ❌ | ✅ | ❌ |
79-
| dpn131 | imagenet | 76M | ❌ | ✅ | ❌ |
80-
| vgg11 | imagenet | 9M | ✅ | ✅ | ✅ |
81-
| vgg11_bn | imagenet | 9M | ✅ | ✅ | ✅ |
82-
| vgg13 | imagenet | 9M | ✅ | ✅ | ✅ |
83-
| vgg13_bn | imagenet | 9M | ✅ | ✅ | ✅ |
84-
| vgg16 | imagenet | 14M | ✅ | ✅ | ✅ |
85-
| vgg16_bn | imagenet | 14M | ✅ | ✅ | ✅ |
86-
| vgg19 | imagenet | 20M | ✅ | ✅ | ✅ |
87-
| vgg19_bn | imagenet | 20M | ✅ | ✅ | ✅ |
88-
| senet154 | imagenet | 113M | ✅ | ✅ | ✅ |
89-
| se_resnet50 | imagenet | 26M | ✅ | ✅ | ✅ |
90-
| se_resnet101 | imagenet | 47M | ✅ | ✅ | ✅ |
91-
| se_resnet152 | imagenet | 64M | ✅ | ✅ | ✅ |
92-
| se_resnext50_32x4d | imagenet | 25M | ✅ | ✅ | ✅ |
93-
| se_resnext101_32x4d | imagenet | 46M | ✅ | ✅ | ✅ |
94-
| densenet121 | imagenet | 6M | ✅ | ✅ | ✅ |
95-
| densenet169 | imagenet | 12M | ✅ | ✅ | ✅ |
96-
| densenet201 | imagenet | 18M | ✅ | ✅ | ✅ |
97-
| densenet161 | imagenet | 26M | ✅ | ✅ | ✅ |
98-
| inceptionresnetv2 | imagenet<br>imagenet+background | 54M | ✅ | ✅ | ✅ |
99-
| inceptionv4 | imagenet<br>imagenet+background | 41M | ✅ | ✅ | ✅ |
100-
| efficientnet-b0 | imagenet<br>advprop | 4M | ❌ | ❌ | ✅ |
101-
| efficientnet-b1 | imagenet<br>advprop | 6M | ❌ | ❌ | ✅ |
102-
| efficientnet-b2 | imagenet<br>advprop | 7M | ❌ | ❌ | ✅ |
103-
| efficientnet-b3 | imagenet<br>advprop | 10M | ❌ | ❌ | ✅ |
104-
| efficientnet-b4 | imagenet<br>advprop | 17M | ❌ | ❌ | ✅ |
105-
| efficientnet-b5 | imagenet<br>advprop | 28M | ❌ | ❌ | ✅ |
106-
| efficientnet-b6 | imagenet<br>advprop | 40M | ❌ | ❌ | ✅ |
107-
| efficientnet-b7 | imagenet<br>advprop | 63M | ❌ | ❌ | ✅ |
108-
| mobilenet_v2 | imagenet | 2M | ✅ | ✅ | ✅ |
109-
| xception | imagenet | 20M | ✅ | ✅ | ✅ |
110-
| timm-efficientnet-b0 | imagenet<br>advprop<br>noisy-student | 4M | ✅ | ✅ | ✅ |
111-
| timm-efficientnet-b1 | imagenet<br>advprop<br>noisy-student | 6M | ✅ | ✅ | ✅ |
112-
| timm-efficientnet-b2 | imagenet<br>advprop<br>noisy-student | 7M | ✅ | ✅ | ✅ |
113-
| timm-efficientnet-b3 | imagenet<br>advprop<br>noisy-student | 10M | ✅ | ✅ | ✅ |
114-
| timm-efficientnet-b4 | imagenet<br>advprop<br>noisy-student | 17M | ✅ | ✅ | ✅ |
115-
| timm-efficientnet-b5 | imagenet<br>advprop<br>noisy-student | 28M | ✅ | ✅ | ✅ |
116-
| timm-efficientnet-b6 | imagenet<br>advprop<br>noisy-student | 40M | ✅ | ✅ | ✅ |
117-
| timm-efficientnet-b7 | imagenet<br>advprop<br>noisy-student | 63M | ✅ | ✅ | ✅ |
118-
| timm-efficientnet-b8 | imagenet<br>advprop | 84M | ✅ | ✅ | ✅ |
119-
| timm-efficientnet-l2 | noisy-student<br>noisy-student-475 | 474M | ✅ | ✅ | ✅ |
120-
| timm-tf_efficientnet_lite0 | imagenet | 3M | ✅ | ✅ | ✅ |
121-
| timm-tf_efficientnet_lite1 | imagenet | 4M | ✅ | ✅ | ✅ |
122-
| timm-tf_efficientnet_lite2 | imagenet | 4M | ✅ | ✅ | ✅ |
123-
| timm-tf_efficientnet_lite3 | imagenet | 6M | ✅ | ✅ | ✅ |
124-
| timm-tf_efficientnet_lite4 | imagenet | 11M | ✅ | ✅ | ✅ |
125-
| timm-skresnet18 | imagenet | 11M | ✅ | ✅ | ✅ |
126-
| timm-skresnet34 | imagenet | 21M | ✅ | ✅ | ✅ |
127-
| timm-skresnext50_32x4d | imagenet | 23M | ✅ | ✅ | ✅ |
128-
| mit_b0 | imagenet | 3M | ✅ | ✅ | ✅ |
129-
| mit_b1 | imagenet | 13M | ✅ | ✅ | ✅ |
130-
| mit_b2 | imagenet | 24M | ✅ | ✅ | ✅ |
131-
| mit_b3 | imagenet | 44M | ✅ | ✅ | ✅ |
132-
| mit_b4 | imagenet | 60M | ✅ | ✅ | ✅ |
133-
| mit_b5 | imagenet | 81M | ✅ | ✅ | ✅ |
134-
| mobileone_s0 | imagenet | 4M | ✅ | ✅ | ✅ |
135-
| mobileone_s1 | imagenet | 3M | ✅ | ✅ | ✅ |
136-
| mobileone_s2 | imagenet | 5M | ✅ | ✅ | ✅ |
137-
| mobileone_s3 | imagenet | 8M | ✅ | ✅ | ✅ |
138-
| mobileone_s4 | imagenet | 12M | ✅ | ✅ | ✅ |
139-
+----------------------------+--------------------------------------+-----------+--------+---------+--------+
140-
60+
============================ ==================================== =========== ======== ========= ========
61+
Encoder Pretrained weights Params, M Script Compile Export
62+
============================ ==================================== =========== ======== ========= ========
63+
resnet18 imagenet / ssl / swsl 11M ✅ ✅ ✅
64+
resnet34 imagenet 21M ✅ ✅ ✅
65+
resnet50 imagenet / ssl / swsl 23M ✅ ✅ ✅
66+
resnet101 imagenet 42M ✅ ✅ ✅
67+
resnet152 imagenet 58M ✅ ✅ ✅
68+
resnext50_32x4d imagenet / ssl / swsl 22M ✅ ✅ ✅
69+
resnext101_32x4d ssl / swsl 42M ✅ ✅ ✅
70+
resnext101_32x8d imagenet / instagram / ssl / swsl 86M ✅ ✅ ✅
71+
resnext101_32x16d instagram / ssl / swsl 191M ✅ ✅ ✅
72+
resnext101_32x32d instagram 466M ✅ ✅ ✅
73+
resnext101_32x48d instagram 826M ✅ ✅ ✅
74+
dpn68 imagenet 11M ❌ ✅ ✅
75+
dpn68b imagenet+5k 11M ❌ ✅ ✅
76+
dpn92 imagenet+5k 34M ❌ ✅ ✅
77+
dpn98 imagenet 58M ❌ ✅ ✅
78+
dpn107 imagenet+5k 84M ❌ ✅ ✅
79+
dpn131 imagenet 76M ❌ ✅ ✅
80+
vgg11 imagenet 9M ✅ ✅ ✅
81+
vgg11_bn imagenet 9M ✅ ✅ ✅
82+
vgg13 imagenet 9M ✅ ✅ ✅
83+
vgg13_bn imagenet 9M ✅ ✅ ✅
84+
vgg16 imagenet 14M ✅ ✅ ✅
85+
vgg16_bn imagenet 14M ✅ ✅ ✅
86+
vgg19 imagenet 20M ✅ ✅ ✅
87+
vgg19_bn imagenet 20M ✅ ✅ ✅
88+
senet154 imagenet 113M ✅ ✅ ✅
89+
se_resnet50 imagenet 26M ✅ ✅ ✅
90+
se_resnet101 imagenet 47M ✅ ✅ ✅
91+
se_resnet152 imagenet 64M ✅ ✅ ✅
92+
se_resnext50_32x4d imagenet 25M ✅ ✅ ✅
93+
se_resnext101_32x4d imagenet 46M ✅ ✅ ✅
94+
densenet121 imagenet 6M ✅ ✅ ✅
95+
densenet169 imagenet 12M ✅ ✅ ✅
96+
densenet201 imagenet 18M ✅ ✅ ✅
97+
densenet161 imagenet 26M ✅ ✅ ✅
98+
inceptionresnetv2 imagenet / imagenet+background 54M ✅ ✅ ✅
99+
inceptionv4 imagenet / imagenet+background 41M ✅ ✅ ✅
100+
efficientnet-b0 imagenet / advprop 4M ❌ ❌ ✅
101+
efficientnet-b1 imagenet / advprop 6M ❌ ❌ ✅
102+
efficientnet-b2 imagenet / advprop 7M ❌ ❌ ✅
103+
efficientnet-b3 imagenet / advprop 10M ❌ ❌ ✅
104+
efficientnet-b4 imagenet / advprop 17M ❌ ❌ ✅
105+
efficientnet-b5 imagenet / advprop 28M ❌ ❌ ✅
106+
efficientnet-b6 imagenet / advprop 40M ❌ ❌ ✅
107+
efficientnet-b7 imagenet / advprop 63M ❌ ❌ ✅
108+
mobilenet_v2 imagenet 2M ✅ ✅ ✅
109+
xception imagenet 20M ✅ ✅ ✅
110+
timm-efficientnet-b0 imagenet / advprop / noisy-student 4M ✅ ✅ ✅
111+
timm-efficientnet-b1 imagenet / advprop / noisy-student 6M ✅ ✅ ✅
112+
timm-efficientnet-b2 imagenet / advprop / noisy-student 7M ✅ ✅ ✅
113+
timm-efficientnet-b3 imagenet / advprop / noisy-student 10M ✅ ✅ ✅
114+
timm-efficientnet-b4 imagenet / advprop / noisy-student 17M ✅ ✅ ✅
115+
timm-efficientnet-b5 imagenet / advprop / noisy-student 28M ✅ ✅ ✅
116+
timm-efficientnet-b6 imagenet / advprop / noisy-student 40M ✅ ✅ ✅
117+
timm-efficientnet-b7 imagenet / advprop / noisy-student 63M ✅ ✅ ✅
118+
timm-efficientnet-b8 imagenet / advprop 84M ✅ ✅ ✅
119+
timm-efficientnet-l2 noisy-student / noisy-student-475 474M ✅ ✅ ✅
120+
timm-tf_efficientnet_lite0 imagenet 3M ✅ ✅ ✅
121+
timm-tf_efficientnet_lite1 imagenet 4M ✅ ✅ ✅
122+
timm-tf_efficientnet_lite2 imagenet 4M ✅ ✅ ✅
123+
timm-tf_efficientnet_lite3 imagenet 6M ✅ ✅ ✅
124+
timm-tf_efficientnet_lite4 imagenet 11M ✅ ✅ ✅
125+
timm-skresnet18 imagenet 11M ✅ ✅ ✅
126+
timm-skresnet34 imagenet 21M ✅ ✅ ✅
127+
timm-skresnext50_32x4d imagenet 23M ✅ ✅ ✅
128+
mit_b0 imagenet 3M ✅ ✅ ✅
129+
mit_b1 imagenet 13M ✅ ✅ ✅
130+
mit_b2 imagenet 24M ✅ ✅ ✅
131+
mit_b3 imagenet 44M ✅ ✅ ✅
132+
mit_b4 imagenet 60M ✅ ✅ ✅
133+
mit_b5 imagenet 81M ✅ ✅ ✅
134+
mobileone_s0 imagenet 4M ✅ ✅ ✅
135+
mobileone_s1 imagenet 3M ✅ ✅ ✅
136+
mobileone_s2 imagenet 5M ✅ ✅ ✅
137+
mobileone_s3 imagenet 8M ✅ ✅ ✅
138+
mobileone_s4 imagenet 12M ✅ ✅ ✅
139+
============================ ==================================== =========== ======== ========= ========

0 commit comments

Comments
 (0)