torchbenchmark/models/timm_efficientdet/metadata.yaml (9 lines of code) (raw):

eval_benchmark: false eval_deterministic: false eval_nograd: true optimized_for_inference: false train_benchmark: false train_deterministic: false not_implemented: # Disable CUDA test because of insufficient GPU memory on CI machine - device: cuda # CPU is not supported - device: cpu