From f60b9688cf60a5c9674de9eff195a69aec988ad4 Mon Sep 17 00:00:00 2001
From: Zhiyi Li <86362692+zhiyil-graphcore@users.noreply.github.com>
Date: Fri, 28 Jul 2023 14:55:04 +0100
Subject: [PATCH] Update pcqm4m_ipu.yaml

---
 expts/hydra-configs/dataset/accelerator/pcqm4m_ipu.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/expts/hydra-configs/dataset/accelerator/pcqm4m_ipu.yaml b/expts/hydra-configs/dataset/accelerator/pcqm4m_ipu.yaml
index 31cba7fc3..6502f9414 100644
--- 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