add set_global_seed in main.py

Author: dayyass

config.yaml

@@ -1,5 +1,6 @@
 torch:
   device: 'cpu'
+  seed: 42
 
 prepare_data:
   train_data:

main.py

@@ -1,3 +1,4 @@
+import os
 from argparse import ArgumentParser
 from collections import Counter
 
@@ -18,15 +19,20 @@
 )
 from pytorch_ner.save import save_model
 from pytorch_ner.train import train
-from pytorch_ner.utils import str_to_class
+from pytorch_ner.utils import set_global_seed, str_to_class
 
 
 def main(path_to_config: str):
 
     with open(path_to_config, mode="r") as fp:
         config = yaml.safe_load(fp)
 
+    # check existence of save path_to_folder
+    if os.path.exists(config["save"]["path_to_folder"]):
+        raise FileExistsError("save directory already exists")
+
     device = torch.device(config["torch"]["device"])
+    set_global_seed(config["torch"]["seed"])
 
     # LOAD DATA
 

pytorch_ner/save.py

@@ -18,7 +18,7 @@ def save_model(
     config: Dict,
     export_onnx: bool = False,
 ):
-    # make empty dir
+    # check existence of save path_to_folder
     if os.path.exists(path_to_folder):
         raise FileExistsError("save directory already exists")
     mkdir(path_to_folder)