Enable sam_fast

pytorch · Jan 29, 2025 · d158caf · d158caf
1 parent 887905b
commit d158caf
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/torchbenchmark/models/sam_fast/metadata.yaml b/torchbenchmark/models/sam_fast/metadata.yaml
@@ -7,11 +7,7 @@ eval_nograd: true
 train_benchmark: false
 train_deterministic: false
 not_implemented:
-- device: cpu
-- device: cuda
-  test: example
-# eval test exceeds 300s time limit on A10G
-# https://github.com/pytorch/benchmark/actions/runs/8972426901/job/24640395990
-- device: NVIDIA A10G
-  test: eval
+  - device: cpu
+  - device: cuda
+    test: example
 skip_cuda_memory_leak: true