runtime_args.py

'''Configurations
'''

import argparse

parser = argparse.ArgumentParser()

parser.add_argument('--data_folder', type=str, help='Specify the path to the folder where the data is.', required=True)
parser.add_argument('--graphs_folder', type=str, help='Specify the path to the folder where the graphs should be written to.', default='./graphs/')
parser.add_argument('--use_cbam', help='Use this flag in order to make use of CBAM.', action='store_true')
parser.add_argument('--model', type=str, help='Specify the model to train. Either resnet-50 or resnet-50-cbam.', default='resnet-50')
parser.add_argument('--epoch', type=int, help='Specify the number of epochs for the training.', default=50)
parser.add_argument('--batch_size', type=int, help='Specify the batch size to be used during training/testing.', default=10)
parser.add_argument('--num_classes', type=int, help='Specify the number of classes the dataset has.', default=6)
parser.add_argument('--learning_rate', type=float, help='Specify the learning rate to be used during training.', default=1e-4)
parser.add_argument('--decay_rate', type=float, help='Specify the decay rate to apply to the learning rate to be used during training.', default=0.98)
parser.add_argument('--num_workers', type=int, help='Specify the number of workers to be used to load the data.', default=4)
parser.add_argument('--shuffle', type=bool, help='Specify if the data for training/testing should be shuffled or not.', default=True)
parser.add_argument('--img_size', type=int, help='Specify the size of the input image.', default=224)
parser.add_argument('--gpus', type=int, help='Specify the number of GPU in a node.', default=1)
parser.add_argument('--nodes', type=int, help='Specify the number of nodes in total.', default=1)
parser.add_argument('--rank', type=int, help='Specify the rank of the node.', default=0)
parser.add_argument('--img_depth', type=int, help='Specify the depth of the input image.', default=3)
parser.add_argument('--device', type=str, help='Specify which device to be used for the evaluation. Either "cpu" or "gpu".', default='gpu')

args = parser.parse_args()