diff --git a/sotabench.py b/sotabench.py index 4d471155..3e273182 100644 --- a/sotabench.py +++ b/sotabench.py @@ -135,7 +135,7 @@ model_list = [ _entry('seresnext50_32x4d', 'SE-ResNeXt-50 32x4d', '1709.01507'), _entry('seresnext101_32x4d', 'SE-ResNeXt-101 32x4d', '1709.01507'), _entry('senet154', 'SENet-154', '1709.01507'), - _entry('xception', 'Xception', '1610.02357'), + _entry('xception', 'Xception', '1610.02357', batch_size=BATCH_SIZE//2), ## Torchvision weights # _entry('densenet121'),