Graphcore
/

t5-small-ipu

Model card Files Files and versions Community

jimypbr commited on Mar 14, 2022

Commit

67e1552

•

1 Parent(s): 6fccd15

Update ipu_config.json

Files changed (1) hide show

ipu_config.json +2 -2

ipu_config.json CHANGED Viewed

@@ -8,8 +8,8 @@
   "device_iterations": 1,
   "inference_device_iterations": 5,
-  "replication_factor": 4,
-  "inference_replication_factor": 4,
   "gradient_accumulation_steps": 512,
   "executable_cache_dir": "./exe_cache",

   "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": 512,
   "executable_cache_dir": "./exe_cache",