@@ -117,11 +117,7 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
117117
118118 _entry ('wide_resnet50_2' , 'Wide-ResNet-50' , '1605.07146' ),
119119
120- _entry ('seresnet18' , 'SE-ResNet-18' , '1709.01507' ),
121- _entry ('seresnet34' , 'SE-ResNet-34' , '1709.01507' ),
122120 _entry ('seresnet50' , 'SE-ResNet-50' , '1709.01507' ),
123- _entry ('seresnext26_32x4d' , 'SE-ResNeXt-26 32x4d' , '1709.01507' ,
124- model_desc = 'Block cfg of SE-ResNeXt-34 w/ Bottleneck' ),
125121 _entry ('seresnext26d_32x4d' , 'SE-ResNeXt-26-D 32x4d' , '1812.01187' ,
126122 model_desc = 'Block cfg of SE-ResNeXt-34 w/ Bottleneck, deep stem, and avg-pool in downsample layers.' ),
127123 _entry ('seresnext26t_32x4d' , 'SE-ResNeXt-26-T 32x4d' , '1812.01187' ,
@@ -261,9 +257,13 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
261257 _entry ('nasnetalarge' , 'NASNet-A Large' , '1707.07012' , batch_size = BATCH_SIZE // 4 ),
262258 _entry ('pnasnet5large' , 'PNASNet-5' , '1712.00559' , batch_size = BATCH_SIZE // 4 ),
263259 _entry ('xception' , 'Xception' , '1610.02357' , batch_size = BATCH_SIZE // 2 ),
260+ _entry ('legacy_seresnet18' , 'SE-ResNet-18' , '1709.01507' ),
261+ _entry ('legacy_seresnet34' , 'SE-ResNet-34' , '1709.01507' ),
264262 _entry ('legacy_seresnet50' , 'SE-ResNet-50' , '1709.01507' ),
265263 _entry ('legacy_seresnet101' , 'SE-ResNet-101' , '1709.01507' ),
266264 _entry ('legacy_seresnet152' , 'SE-ResNet-152' , '1709.01507' ),
265+ _entry ('legacy_seresnext26_32x4d' , 'SE-ResNeXt-26 32x4d' , '1709.01507' ,
266+ model_desc = 'Block cfg of SE-ResNeXt-34 w/ Bottleneck' ),
267267 _entry ('legacy_seresnext50_32x4d' , 'SE-ResNeXt-50 32x4d' , '1709.01507' ),
268268 _entry ('legacy_seresnext101_32x4d' , 'SE-ResNeXt-101 32x4d' , '1709.01507' ),
269269 _entry ('legacy_senet154' , 'SENet-154' , '1709.01507' ),
@@ -522,6 +522,9 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
522522 evaluator .add (dict (zip (image_ids , list (output ))))
523523 sample_count += num_samples
524524 bar .update (num_samples )
525+ if evaluator .cache_exists :
526+ break
527+
525528 bar .close ()
526529
527530 evaluator .save ()
0 commit comments