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: