Graphcore
/

bart-base-ipu

Model card Files Files and versions Community

Jinchen commited on Jul 28, 2022

Commit

3995067

•

1 Parent(s): a09a542

Update ipu_config.json

Files changed (1) hide show

ipu_config.json +1 -1

ipu_config.json CHANGED Viewed

@@ -14,6 +14,6 @@
     "executable_cache_dir": "./exe_cache",
     "ipus_per_replica": 4,
-    "layers_per_ipu": [0, 4, 4, 4],
     "matmul_proportion": [0.25, 0.25, 0.25, 0.25]
 }

     "executable_cache_dir": "./exe_cache",
     "ipus_per_replica": 4,
+    "layers_per_ipu": [0, 6, 3, 3],
     "matmul_proportion": [0.25, 0.25, 0.25, 0.25]
 }