config.py

import argparse

def parse_opts():

    parser = argparse.ArgumentParser()
    
    #Paths
    parser.add_argument('--save_dir', default='./output/', type=str, help='Where to save training outputs.')
    parser.add_argument("--movie_path", type=str, default="./TestResults/", help="Path of video for containers, thumbnails, and segments for testing")
    parser.add_argument("--checkpoint_model", type=str, help="Optional path to checkpoint model for testing")
    
    # Dataset
    parser.add_argument('--dataset_path', default='./data/ucf101/', type=str, help='Path to location of dataset images')
    parser.add_argument('--dataset', default='ucf101', type=str, help='Dataset string (ucf13 | ucf101 | hmdb51)')
    parser.add_argument('--num_classes', default=101, type=int, help= 'Number of classes (ucf13: 13, ucf101: 101, hmdb51: 51)')

    # Preprocessing pipeline
    parser.add_argument('--spatial_size', default= 224, type=int, help='Height and width and hight of inputs Inception 299| efficentNet 244')

    # Models (general)
    parser.add_argument('--model', default='efficientNet', type=str, help='( inceptionV3 | efficientNet | resnext | densenet)')
    parser.add_argument('--manual_seed', default=1, type=int, help='Manually set random seed')

    # Optimization
    parser.add_argument('--early_stopping_patience', default=10, type=int, help='Early stopping patience (number of epochs)')
    parser.add_argument('--batch_size', default=32, type=int, help='Batch Size')
    parser.add_argument('--num_epochs', default=10, type=int, help='Number of epochs to train for')
    parser.add_argument('--learning_rate', default=0.001, type=float, help='Initial learning rate (divided by 10 while training by lr-scheduler)')    
    
    #Misc
    parser.add_argument('--device', default='1',type=str, help='GPU device string number 1 |1,2')
    parser.add_argument('--walter_ip', default='http://115.145.67.104/dataset/movies/', type=str, help='Server IP address')
    parser.add_argument('--genre', default='wt',type=str, help='Movie genre fb: footbal | ac: action | wt: western | cr: cricket')

    return parser.parse_args()