File size: 567 Bytes
265a281 4cd6d3b 265a281 f523668 265a281 |
1 2 3 4 5 6 7 8 9 10 11 12 |
{
"optimizer_state_offchip": true,
"replicated_tensor_sharding": true,
"enable_half_partials": true,
"gradient_accumulation_steps": 32,
"executable_cache_dir": "./exe_cache",
"layers_per_ipu": [1, 2, 3, 3, 3, 0, 6, 6],
"matmul_proportion": [0.6, 0.6, 0.2, 0.2, 0.2, 0.6, 0.2, 0.2],
"serialized_projection_splits_per_ipu": [0, 0, 0, 0, 0, 4, 0, 0]
} |