@@ -116,6 +116,7 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
116
116
model_desc = 'Block cfg of SE-ResNeXt-34 w/ Bottleneck, deep tiered narrow stem, and avg-pool in downsample layers.' ),
117
117
_entry ('spnasnet_100' , 'Single-Path NAS' , '1904.02877' ,
118
118
model_desc = 'Trained in PyTorch with SGD, cosine LR decay' ),
119
+
119
120
_entry ('tf_efficientnet_b0' , 'EfficientNet-B0 (AutoAugment)' , '1905.11946' ,
120
121
model_desc = 'Ported from official Google AI Tensorflow weights' ),
121
122
_entry ('tf_efficientnet_b1' , 'EfficientNet-B1 (AutoAugment)' , '1905.11946' ,
@@ -134,6 +135,7 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
134
135
model_desc = 'Ported from official Google AI Tensorflow weights' ),
135
136
_entry ('tf_efficientnet_b8' , 'EfficientNet-B8 (RandAugment)' , '1905.11946' , batch_size = BATCH_SIZE // 8 ,
136
137
model_desc = 'Ported from official Google AI Tensorflow weights' ),
138
+
137
139
_entry ('tf_efficientnet_b0_ap' , 'EfficientNet-B0 (AdvProp)' , '1911.09665' ,
138
140
model_desc = 'Ported from official Google AI Tensorflow weights' ),
139
141
_entry ('tf_efficientnet_b1_ap' , 'EfficientNet-B1 (AdvProp)' , '1911.09665' ,
@@ -152,6 +154,28 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
152
154
model_desc = 'Ported from official Google AI Tensorflow weights' ),
153
155
_entry ('tf_efficientnet_b8_ap' , 'EfficientNet-B8 (AdvProp)' , '1911.09665' , batch_size = BATCH_SIZE // 8 ,
154
156
model_desc = 'Ported from official Google AI Tensorflow weights' ),
157
+
158
+ _entry ('tf_efficientnet_b0_ns' , 'EfficientNet-B0 (NoisyStudent)' , '1911.04252' ,
159
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
160
+ _entry ('tf_efficientnet_b1_ns' , 'EfficientNet-B1 (NoisyStudent)' , '1911.04252' ,
161
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
162
+ _entry ('tf_efficientnet_b2_ns' , 'EfficientNet-B2 (NoisyStudent)' , '1911.04252' ,
163
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
164
+ _entry ('tf_efficientnet_b3_ns' , 'EfficientNet-B3 (NoisyStudent)' , '1911.04252' , batch_size = BATCH_SIZE // 2 ,
165
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
166
+ _entry ('tf_efficientnet_b4_ns' , 'EfficientNet-B4 (NoisyStudent)' , '1911.04252' , batch_size = BATCH_SIZE // 2 ,
167
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
168
+ _entry ('tf_efficientnet_b5_ns' , 'EfficientNet-B5 (NoisyStudent)' , '1911.04252' , batch_size = BATCH_SIZE // 4 ,
169
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
170
+ _entry ('tf_efficientnet_b6_ns' , 'EfficientNet-B6 (NoisyStudent)' , '1911.04252' , batch_size = BATCH_SIZE // 8 ,
171
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
172
+ _entry ('tf_efficientnet_b7_ns' , 'EfficientNet-B7 (NoisyStudent)' , '1911.04252' , batch_size = BATCH_SIZE // 8 ,
173
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
174
+ _entry ('tf_efficientnet_l2_ns_475' , 'EfficientNet-L2 475 (NoisyStudent)' , '1911.04252' , batch_size = BATCH_SIZE // 16 ,
175
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
176
+ _entry ('tf_efficientnet_l2_ns' , 'EfficientNet-L2 (NoisyStudent)' , '1911.04252' , batch_size = BATCH_SIZE // 64 ,
177
+ model_desc = 'Ported from official Google AI Tensorflow weights' ),
178
+
155
179
_entry ('tf_efficientnet_cc_b0_4e' , 'EfficientNet-CondConv-B0 4 experts' , '1904.04971' ,
156
180
model_desc = 'Ported from official Google AI Tensorflow weights' ),
157
181
_entry ('tf_efficientnet_cc_b0_8e' , 'EfficientNet-CondConv-B0 8 experts' , '1904.04971' ,
0 commit comments