LightChen2333 commited on
Commit
9f2b015
·
1 Parent(s): 328a7a1

Delete run.py

Browse files
Files changed (1) hide show
  1. run.py +0 -34
run.py DELETED
@@ -1,34 +0,0 @@
1
- '''
2
- Author: Qiguang Chen
3
- Date: 2023-01-11 10:39:26
4
- LastEditors: Qiguang Chen
5
- LastEditTime: 2023-02-19 19:01:40
6
- Description: main executive file
7
-
8
- '''
9
- import argparse
10
-
11
- from common.config import Config
12
- from common.model_manager import ModelManager
13
-
14
-
15
- def main():
16
- parser = argparse.ArgumentParser()
17
- parser.add_argument('--config_path', '-cp', type=str, default='config/stack-propagation.yaml')
18
- parser.add_argument('--model', '-m', type=str, default=None)
19
- parser.add_argument('--dataset', '-ds', type=str, default=None)
20
- parser.add_argument('--device', '-dv', type=str, default=None)
21
- parser.add_argument('--learning_rate', '-lr', type=float, default=None)
22
- parser.add_argument('--epoch_num', '-en', type=int, default=None)
23
- args = parser.parse_args()
24
- config = Config.load_from_args(args)
25
- model_manager = ModelManager(config)
26
- model_manager.init_model()
27
- if config.base.get("train"):
28
- model_manager.train()
29
- if not config.base.get("train") and config.base.get("test"):
30
- model_manager.test()
31
-
32
-
33
- if __name__ == "__main__":
34
- main()