flpelerin commited on
Commit
4349b9b
·
1 Parent(s): 97e7631

Update 3 files

Browse files

- /util.py
- /trainer.py
- /trainer.cli.py

Files changed (3) hide show
  1. trainer.cli.py +26 -0
  2. trainer.py +3 -0
  3. util.py +18 -0
trainer.cli.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+
3
+
4
+
5
+ from argparse import ArgumentParser
6
+
7
+ from util import ConfigParser
8
+
9
+
10
+
11
+ parser = ArgumentParser(
12
+ prog='Trainer implementation, using Pytorch',
13
+ description=''
14
+ )
15
+
16
+
17
+
18
+ if __name__ == '__main__':
19
+
20
+ parser.add_argument('-p', '--config_path')
21
+ args = parser.parse_args()
22
+
23
+
24
+ parser = ConfigParser(args.config_path)
25
+
26
+ print(parser.config)
trainer.py ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+
2
+ class Trainer:
3
+ def __init__(config: Config):
util.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import json
2
+
3
+
4
+
5
+ class ConfigParser:
6
+ def __init__(path: str):
7
+ with open(path, 'r') as f:
8
+ json_obj = json.load(f)
9
+
10
+ config = json.loads(json_obj, object_hook=lambda x: SimpleNamespace(**x))
11
+
12
+ self.path = path
13
+ self.config = config
14
+ self.json_obj = json_obj
15
+
16
+
17
+
18
+