update metadata.yml for unet1d

pytorch · May 7, 2024 · aa3ea2a · aa3ea2a
1 parent fa42c44
commit aa3ea2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchbenchmark/models/unet1d/metadata.yml b/torchbenchmark/models/unet1d/metadata.yml
@@ -4,5 +4,5 @@ devices:
 eval_benchmark: true
 eval_deterministic: false
 eval_nograd: true
-train_benchmark: true
+train_benchmark: false
 train_deterministic: false