Merge pull request #569 from AMHermansen/config-patch-AMHermansen

Changed dtype in config files
graphnet-team · Aug 21, 2023 · 720aba6 · 720aba6
2 parents f6e0328 + 9702be5
commit 720aba6
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/configs/datasets/dev_lvl7_robustness_muon_neutrino_0000.yml b/configs/datasets/dev_lvl7_robustness_muon_neutrino_0000.yml
@@ -5,7 +5,7 @@ graph_definition:
     detector:
       arguments: {}
       class_name: IceCube86
-    dtype: null
+    dtype: torch.float32
     nb_nearest_neighbours: 8
     node_definition:
       arguments: {}

diff --git a/configs/datasets/test_data_sqlite.yml b/configs/datasets/test_data_sqlite.yml
@@ -5,7 +5,7 @@ graph_definition:
     detector:
       arguments: {}
       class_name: Prometheus
-    dtype: null
+    dtype: torch.float32
     nb_nearest_neighbours: 8
     node_definition:
       arguments: {}

diff --git a/configs/datasets/training_classification_example_data_sqlite.yml b/configs/datasets/training_classification_example_data_sqlite.yml
@@ -5,7 +5,7 @@ graph_definition:
     detector:
       arguments: {}
       class_name: Prometheus
-    dtype: null
+    dtype: torch.float32
     nb_nearest_neighbours: 8
     node_definition:
       arguments: {}

diff --git a/configs/datasets/training_example_data_parquet.yml b/configs/datasets/training_example_data_parquet.yml
@@ -5,7 +5,7 @@ graph_definition:
     detector:
       arguments: {}
       class_name: Prometheus
-    dtype: null
+    dtype: torch.float32
     nb_nearest_neighbours: 8
     node_definition:
       arguments: {}

diff --git a/configs/datasets/training_example_data_sqlite.yml b/configs/datasets/training_example_data_sqlite.yml
@@ -5,7 +5,7 @@ graph_definition:
     detector:
       arguments: {}
       class_name: Prometheus
-    dtype: null
+    dtype: torch.float32
     nb_nearest_neighbours: 8
     node_definition:
       arguments: {}