Graphcore
/

bert-base-ipu

Model card Files Files and versions Community

jimypbr commited on Mar 21, 2022

Commit

6defde9

•

1 Parent(s): 451f8d9

Update ipu_config.json

Files changed (1) hide show

ipu_config.json +2 -2

ipu_config.json CHANGED Viewed

@@ -6,8 +6,8 @@
   "enable_half_first_order_momentum": true,
   "enable_half_partials": true,
-  "device_iterations": 2,
-  "inference_device_iterations": 4,
   "replication_factor": {"pod4": 1, "pod8": 2, "pod16": 4, "pod32": 8, "pod64": 16, "default": 1},
   "inference_replication_factor": {"pod4": 1, "pod8": 2, "pod16": 4, "pod32": 8, "pod64": 16, "default": 1},
   "gradient_accumulation_steps": 16,

   "enable_half_first_order_momentum": true,
   "enable_half_partials": true,
+  "device_iterations": 1,
+  "inference_device_iterations": 5,
   "replication_factor": {"pod4": 1, "pod8": 2, "pod16": 4, "pod32": 8, "pod64": 16, "default": 1},
   "inference_replication_factor": {"pod4": 1, "pod8": 2, "pod16": 4, "pod32": 8, "pod64": 16, "default": 1},
   "gradient_accumulation_steps": 16,