Training in progress, step 500

config.json

@@ -2,7 +2,7 @@
   "_name_or_path": "gpt2",
   "activation_function": "gelu_new",
   "architectures": [
-    "PipelinedGPT2LMHeadModel"
+    "PoptorchPipelinedGPT2LMHeadModel"
   ],
   "attn_pdrop": 0.1,
   "bos_token_id": 50256,

ipu_config.json

@@ -3,11 +3,11 @@
   "device_iterations": 2,
   "embedding_serialization_factor": 4,
   "enable_half_partials": true,
-  "executable_cache_dir": "/tmp/lmcache/",
+  "executable_cache_dir": "/tmp/lmcache4/",
   "execute_encoder_on_cpu_for_generation": false,
   "gradient_accumulation_steps": 16,
   "inference_device_iterations": 5,
-  "inference_replication_factor": 4,
+  "inference_replication_factor": 1,
   "ipus_per_replica": 4,
   "layers_per_ipu": [
     0,
@@ -20,8 +20,8 @@
   "optimum_version": "1.5.1",
   "output_mode": "final",
   "recompute_checkpoint_every_layer": true,
-  "replicated_tensor_sharding": true,
-  "replication_factor": 4,
+  "replicated_tensor_sharding": false,
+  "replication_factor": 1,
   "seed": 42,
   "sharded_execution_for_inference": false,
   "transformers_version": "4.20.1"

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2c198b53f4c8c7a6525d3e27ecb5c7126b9c0916c930ad6cc8b357ec5a5cedb2
+oid sha256:98be7a37645b50ad99a01b1d44d9ab4abd27738bff597b76c13b02edde1f3ed6
 size 261514473

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a773e48dabdb3e3091dec5107367123865bd42e1bcb30cc18ab8017a3fdb8f1d
+oid sha256:fd0e8adb4537e66ff3c6649d9c805d0ef48c7c4c1b248b1512326c7a21ab9ed2
 size 2735