Graphcore
atsyplikhin commited on
Commit
ccf1322
·
1 Parent(s): 93608c2

Create ipu_config.json

Browse files
Files changed (1) hide show
  1. ipu_config.json +17 -0
ipu_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "embedding_serialization_factor": 0,
3
+ "recompute_checkpoint_every_layer": false,
4
+ "optimizer_state_offchip": true,
5
+ "replicated_tensor_sharding": true,
6
+ "enable_half_first_order_momentum": true,
7
+ "enable_half_partials": true,
8
+ "device_iterations": 1,
9
+ "inference_device_iterations": 8,
10
+ "replication_factor": {"pod8": 1, "pod16": 2, "pod32": 4, "pod64": 8, "default": 1},
11
+ "inference_replication_factor": {"pod8": 1, "pod16": 2, "pod32": 4, "pod64": 8, "default": 1},
12
+ "gradient_accumulation_steps": 32,
13
+ "executable_cache_dir": "./exe_cache",
14
+ "ipus_per_replica": 8,
15
+ "layers_per_ipu": [4,5,4,4,4,5,5,4],
16
+ "matmul_proportion": [0.2,0.2,0.1,0.1,0.1,0.1,0.1,1.0]
17
+ }