Graphcore
/

wav2vec2-base-ipu

Model card Files Files and versions Community

jimypbr commited on Aug 1, 2022

Commit

0ca51cb

•

1 Parent(s): a6f5d68

Update ipu_config.json

Files changed (1) hide show

ipu_config.json +2 -2

ipu_config.json CHANGED Viewed

@@ -6,8 +6,8 @@
     "recompute_checkpoint_every_layer": false,
     "device_iterations": 1,
     "inference_device_iterations": 4,
-    "replication_factor": 1,
-    "inference_replication_factor": 1,
     "gradient_accumulation_steps": 32,
     "executable_cache_dir": "./exe_cache",
     "ipus_per_replica": 4,

     "recompute_checkpoint_every_layer": false,
     "device_iterations": 1,
     "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": 32,
     "executable_cache_dir": "./exe_cache",
     "ipus_per_replica": 4,