Train with config

train_config.py

import configparser
import argparse

def train(data_path, model_path, learning_rate, batch_size):
    ...

if __name__ == "__main__":
    argparser = argparse.ArgumentParser()
    argparser.add_argument("--config", type=str, help="path to config file")
    args = argparser.parse_args()

    cfg = configReader
    cfg.read(args.config)
    data_path = cfg["train"]["data_path"]
    model_path = cfg["train"]["model_path"]
    learning_rate cfg["train"].getfloat("learning_rate")
    batch_size = cfg["train"].getint("batch_size")

    train(data_path, model_path, learning_rate, batch_size)