update

Code/main.py

@@ -8,8 +8,8 @@ import models
 from typing import List
 
 
-def run(raw_args):
-	args=_parse_args(raw_args)
+def run(args):
+	#args=_parse_args(raw_args)
 
 
 	#load test and train dataset as well as tokenizers and models...
@@ -64,10 +64,9 @@ def run(raw_args):
 	#(evaluate... is done internally) but could maybe be implemented here to make average over multiple random seeds
 
 
-def _parse_args(raw_args: List[str]) -> argparse.Namespace:
+if __name__ == "__main__":
 	parser = argparse.ArgumentParser()
 
-
 	#Architecture
 	parser.add_argument(
 		"--architecture",
@@ -144,5 +143,6 @@ def _parse_args(raw_args: List[str]) -> argparse.Namespace:
 		type=int,
 		default=64)
 
-
+	args = parser.parse_args()
+	run(args)