Graphcore
/

groupbert-base-uncased

katalinic commited on Oct 13, 2023

Commit

3d5e057

•

1 Parent(s): 9c7c5f7

Update for SDK3.4 (works with 3.3 as well)

Files changed (1) hide show

ipu_config.json CHANGED Viewed

@@ -9,12 +9,12 @@
   "inference_replication_factor": 1,
   "ipus_per_replica": 4,
   "layers_per_ipu": [
-    0,
-    4,
     4,
     4
   ],
-  "matmul_proportion": [0.6, 0.09, 0.1, 0.15],
   "optimizer_state_offchip": true,
   "optimum_version": "1.4.1",
   "output_mode": "final",

   "inference_replication_factor": 1,
   "ipus_per_replica": 4,
   "layers_per_ipu": [
+    1,
+    3,
     4,
     4
   ],
+  "matmul_proportion": [0.6, 0.2, 0.1, 0.1],
   "optimizer_state_offchip": true,
   "optimum_version": "1.4.1",
   "output_mode": "final",