config.py

import argparse
from easydict import EasyDict as edict


def parse_configs():
    parser = argparse.ArgumentParser()
    ####################################################################
    ##############     Model configs            ###################
    ####################################################################
    parser.add_argument('--input_size', type=int, default=224, metavar='INPSIZE',
                        help='The size of the input image')
    parser.add_argument('--checkpoint', type=str, default='', metavar='CPATH',
                        help='the directory of the checkpoint')

    ####################################################################
    ##############     Dataloader and Running configs            #######
    ####################################################################
    parser.add_argument('--name', type=str, default='40k_new', metavar='Name',
                        help='model name')
    parser.add_argument('--batch_size', type=int, default=8)

    ####################################################################
    ##############     Training strategy            ###################
    ####################################################################

    parser.add_argument('--num_epochs', type=int, default=80, metavar='N',
                        help='number of total epochs to run')
    parser.add_argument('--lr', type=float, default=1e-3, metavar='LR',
                        help='initial learning rate')
    parser.add_argument('--train_loss_fr', type=float, default=10, metavar='LF',
                        help='loss plotting frequency in update steps')
    parser.add_argument('--val_loss_fr', type=float, default=100, metavar='VF',
                        help='loss plotting frequency in update steps')
    parser.add_argument('--custom_loss', type=bool, default=True, metavar='CL',
                        help='loss type')
    parser.add_argument('--use_weights', type=bool, default=True, metavar='CW',
                        help='loss type')

    configs = edict(vars(parser.parse_args()))

    return configs