Update pcqm4m_ipu.yaml

datamol-io · Jul 28, 2023 · f60b968 · f60b968
1 parent 83aa388
commit f60b968
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/expts/hydra-configs/dataset/accelerator/pcqm4m_ipu.yaml b/expts/hydra-configs/dataset/accelerator/pcqm4m_ipu.yaml
@@ -11,7 +11,7 @@ datamodule:
       max_num_nodes_per_graph: 30 # valid max nodes: 51, max_edges: 118
       max_num_edges_per_graph: 120
     # Data handling-related
-    batch_size_training: 64
+    batch_size_training: 32
     batch_size_inference: 16
 
 predictor:
@@ -22,4 +22,4 @@ predictor:
 trainer:
   trainer:
     precision: 16-true
-    accumulate_grad_batches: 4
+    accumulate_grad_batches: 2