Update for SDK3.4 (works with 3.3 as well)
Browse files- ipu_config.json +3 -3
ipu_config.json
CHANGED
@@ -9,12 +9,12 @@
|
|
9 |
"inference_replication_factor": 1,
|
10 |
"ipus_per_replica": 4,
|
11 |
"layers_per_ipu": [
|
12 |
-
|
13 |
-
|
14 |
4,
|
15 |
4
|
16 |
],
|
17 |
-
"matmul_proportion": [0.6, 0.
|
18 |
"optimizer_state_offchip": true,
|
19 |
"optimum_version": "1.4.1",
|
20 |
"output_mode": "final",
|
|
|
9 |
"inference_replication_factor": 1,
|
10 |
"ipus_per_replica": 4,
|
11 |
"layers_per_ipu": [
|
12 |
+
1,
|
13 |
+
3,
|
14 |
4,
|
15 |
4
|
16 |
],
|
17 |
+
"matmul_proportion": [0.6, 0.2, 0.1, 0.1],
|
18 |
"optimizer_state_offchip": true,
|
19 |
"optimum_version": "1.4.1",
|
20 |
"output_mode": "final",
|