smaller stochastic depth

mosaicml · Nov 29, 2024 · 25898a8 · 25898a8
1 parent 7824d90
commit 25898a8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/algorithms/algorithm_settings.py b/tests/algorithms/algorithm_settings.py
@@ -207,19 +207,19 @@
         'model': (
             composer_resnet,
             {
-                'model_name': 'resnet50',
+                'model_name': 'resnet18',
                 'num_classes': 2,
             },
         ),
         'dataset': (
             RandomImageDataset,
             {
-                'shape': (3, 224, 224),
+                'shape': (3, 32, 32), 
             },
         ),
         'kwargs': {
             'stochastic_method': 'block',
-            'target_layer_name': 'ResNetBottleneck',
+            'target_layer_name': 'BasicBlock', 
             'drop_rate': 0.2,
             'drop_distribution': 'linear',
             'drop_warmup': '0.0dur',