diff --git a/examples/pytorch_timm.yaml b/examples/pytorch_timm.yaml new file mode 100644 index 00000000..154eba5e --- /dev/null +++ b/examples/pytorch_timm.yaml @@ -0,0 +1,31 @@ +defaults: + - backend: pytorch # default backend + - launcher: process # default launcher + - benchmark: inference # default benchmark + - experiment # inheriting experiment schema + - _self_ # for hydra 1.1 compatibility + - override hydra/job_logging: colorlog # colorful logging + - override hydra/hydra_logging: colorlog # colorful logging + +model: timm/mobilenetv3_large_100.ra_in1k +experiment_name: pytorch_timm +device: cuda + +launcher: + device_isolation: true + +benchmark: + input_shapes: + batch_size: 1 + +hydra: + run: + dir: runs/${experiment_name} + sweep: + dir: sweeps/${experiment_name} + job: + chdir: true + env_set: + OVERRIDE_BENCHMARKS: 1 + CUDA_VISIBLE_DEVICES: 0 + CUDA_DEVICE_ORDER: PCI_BUS_ID