Graphcore
/

bart-base-ipu

Model card Files Files and versions

katalinic commited on Jul 7, 2023

Commit

c40f326

·

1 Parent(s): 3995067

Remove deprecated kwargs

Files changed (1) hide show

ipu_config.json +2 -3

ipu_config.json CHANGED Viewed

@@ -3,13 +3,12 @@
     "recompute_checkpoint_every_layer": true,
     "optimizer_state_offchip": true,
     "replicated_tensor_sharding": true,
-    "enable_half_first_order_momentum": true,
     "enable_half_partials": true,
     "device_iterations": 1,
     "inference_device_iterations": 6,
-    "replication_factor": {"pod4": 1, "pod8": 2, "pod16": 4, "pod32": 8, "pod64": 16, "default": 1},
-    "inference_replication_factor": {"pod4": 1, "pod8": 2, "pod16": 4, "pod32": 8, "pod64": 16, "default": 1},
     "gradient_accumulation_steps": 512,
     "executable_cache_dir": "./exe_cache",

     "recompute_checkpoint_every_layer": true,
     "optimizer_state_offchip": true,
     "replicated_tensor_sharding": true,
     "enable_half_partials": true,
     "device_iterations": 1,
     "inference_device_iterations": 6,
+    "replication_factor": 1,
+    "inference_replication_factor": 1,
     "gradient_accumulation_steps": 512,
     "executable_cache_dir": "./exe_cache",