diff --git a/benchmarks/_templates/simple/dev.yaml b/benchmarks/_templates/simple/dev.yaml index e3aa94673..affcc977f 100644 --- a/benchmarks/_templates/simple/dev.yaml +++ b/benchmarks/_templates/simple/dev.yaml @@ -6,3 +6,5 @@ template: install_group: torch plan: method: per_gpu + tags: + - monogpu diff --git a/benchmarks/_templates/stdout/dev.yaml b/benchmarks/_templates/stdout/dev.yaml index 2b7e75a34..24c7b8131 100644 --- a/benchmarks/_templates/stdout/dev.yaml +++ b/benchmarks/_templates/stdout/dev.yaml @@ -3,7 +3,8 @@ _template: definition: . install-variant: unpinned install_group: torch - + tags: + - monogpu #argv: # --train_batch_size: 32 # --num_epochs: 5 diff --git a/benchmarks/_templates/voir/dev.yaml b/benchmarks/_templates/voir/dev.yaml index e3aa94673..affcc977f 100644 --- a/benchmarks/_templates/voir/dev.yaml +++ b/benchmarks/_templates/voir/dev.yaml @@ -6,3 +6,5 @@ template: install_group: torch plan: method: per_gpu + tags: + - monogpu