diff --git a/torchbenchmark/models/squeezenet1_1/metadata.yaml b/torchbenchmark/models/squeezenet1_1/metadata.yaml
index a3d0a103f6..b2f17b691c 100644
--- a/torchbenchmark/models/squeezenet1_1/metadata.yaml
+++ b/torchbenchmark/models/squeezenet1_1/metadata.yaml
@@ -5,6 +5,9 @@ eval_benchmark: true
 eval_deterministic: false
 eval_nograd: true
 not_implemented:
-- test: train
+  - device: cpu
+    test: train
+  - device: cuda
+    test: train
 train_benchmark: true
 train_deterministic: false