Merge pull request #3 from SpiritedAwayCN/artifact

modified gen ckpt script
Hsword · Feb 22, 2024 · 832b29f · 832b29f
2 parents 252db55 + 96c0523
commit 832b29f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ckpt/generate_random_gpt_ckpt.py b/ckpt/generate_random_gpt_ckpt.py
@@ -52,6 +52,9 @@ def gen_ckpt(output_dir):
         (hidden_units,),
         (hidden_units,)
     ]
+
+    if not os.path.exists(output_dir):
+        os.mkdir(output_dir)
 
     for shape, name in zip(size_non_block, non_block_name):
         a = np.random.normal(size=shape).astype(np.float32)