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