From 755a78ef4de2c0cde12fa7b25a09422003dae4da Mon Sep 17 00:00:00 2001 From: Eric Wulff Date: Wed, 25 Oct 2023 08:05:55 -0400 Subject: [PATCH] fix config files --- parameters/pyg-clic.yaml | 2 +- parameters/pyg-cms.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/parameters/pyg-clic.yaml b/parameters/pyg-clic.yaml index 3bba81e05..26fe0a201 100644 --- a/parameters/pyg-clic.yaml +++ b/parameters/pyg-clic.yaml @@ -1,7 +1,7 @@ backend: pytorch dataset: clic -data_dir: /mnt/ceph/users/ewulff/tensorflow_datasets/clusters +data_dir: gpus: "0" gpu_batch_multiplier: 1 load: diff --git a/parameters/pyg-cms.yaml b/parameters/pyg-cms.yaml index cff86ebcb..5967196a5 100644 --- a/parameters/pyg-cms.yaml +++ b/parameters/pyg-cms.yaml @@ -1,7 +1,7 @@ backend: pytorch dataset: cms -data_dir: /mnt/ceph/users/ewulff/tensorflow_datasets/ +data_dir: gpus: "0" gpu_batch_multiplier: 1 load: