hmellor commited on
Commit
f523668
1 Parent(s): 265a281

Update ipu_config.json

Browse files
Files changed (1) hide show
  1. ipu_config.json +3 -3
ipu_config.json CHANGED
@@ -6,7 +6,7 @@
6
  "gradient_accumulation_steps": 16,
7
  "executable_cache_dir": "./exe_cache",
8
 
9
- "layers_per_ipu": [4, 4],
10
- "matmul_proportion": 0.6,
11
- "projection_serialization_factor": 4
12
  }
 
6
  "gradient_accumulation_steps": 16,
7
  "executable_cache_dir": "./exe_cache",
8
 
9
+ "layers_per_ipu": [1, 2, 3, 3, 3, 0, 6, 6],
10
+ "matmul_proportion": [0.6, 0.6, 0.2, 0.2, 0.2, 0.6, 0.2, 0.2],
11
+ "serialized_projection_splits_per_ipu": [0, 0, 0, 0, 0, 4, 0, 0]
12
  }