Spaces:
Runtime error
Runtime error
Commit
·
9f2b015
1
Parent(s):
328a7a1
Delete run.py
Browse files
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()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|