diff --git a/torchbenchmark/models/timm_nfnet/metadata.yaml b/torchbenchmark/models/timm_nfnet/metadata.yaml index 95dfc282e4..6f82f648e2 100644 --- a/torchbenchmark/models/timm_nfnet/metadata.yaml +++ b/torchbenchmark/models/timm_nfnet/metadata.yaml @@ -5,7 +5,7 @@ eval_benchmark: true eval_deterministic: false eval_nograd: true not_implemented: -- evice: cuda +- device: cuda test: train train_benchmark: true train_deterministic: false