jimypbr commited on
Commit
5c1732c
1 Parent(s): 869178e

Update ipu_config.json

Browse files
Files changed (1) hide show
  1. ipu_config.json +3 -1
ipu_config.json CHANGED
@@ -9,9 +9,11 @@
9
  "replication_factor": 4,
10
  "gradient_accumulation_steps": 16,
11
  "device_iterations": 1,
 
 
12
  "executable_cache_dir": "./exe_cache",
13
 
14
  "ipus_per_replica": 4,
15
  "layers_per_ipu": [3, 7, 7, 7],
16
  "matmul_proportion": [0.15, 0.18, 0.20, 0.25]
17
- }
 
9
  "replication_factor": 4,
10
  "gradient_accumulation_steps": 16,
11
  "device_iterations": 1,
12
+ "inference_device_iterations": 4,
13
+ "inference_replication_factor": 4,
14
  "executable_cache_dir": "./exe_cache",
15
 
16
  "ipus_per_replica": 4,
17
  "layers_per_ipu": [3, 7, 7, 7],
18
  "matmul_proportion": [0.15, 0.18, 0.20, 0.25]
19
+ }