config.py

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--batch_size", type=int, default=10, help="input batch size")
parser.add_argument("-g", "--gpu", type=int, default=0)
parser.add_argument("-nEpochs", "--epochs", type=int, default=100)
parser.add_argument("-lr", "--learning_rate", type=float, default=0.001)
parser.add_argument(
    "-opt", "--optimizer", type=str, default="adam", choices=("sgd", "adam")
)
parser.add_argument("-dr", "--dataset_path", type=str, default="gta5_new/")
parser.add_argument("-or", "--output_path", type=str, default="gta5_new/")
parser.add_argument("-msp", "--ModelSavePath", type=str, default="models/")
parser.add_argument("-mp", "--ModelPath", type=str, default="gta5_new/Best_Model.pkl")
parser.add_argument("-vs", "--validation_split", type=float, default=0.2)
args = parser.parse_args()