def parse_arguments()

in main_stica.py [0:0]

• 94 lines of code
• 3 McCabe index (conditional complexity)

def parse_arguments():
    def str2bool(v):
        v = v.lower()
        if v in ('yes', 'true', 't', '1'):
            return True
        elif v in ('no', 'false', 'f', '0'):
            return False
        raise ValueError('Boolean argument needs to be true or false. '
                        'Instead, it is %s.' % v)

    parser = argparse.ArgumentParser(description='StiCa')
    parser.register('type', 'bool', str2bool)

    #########################
    #### data parameters ####
    #########################
    
    parser.add_argument('--dataset_name', type=str, default='kinetics',
                        help='name of dataset e.g kinetics, kinetics600')
    parser.add_argument('--root_dir', type=str, default=None,
                        help='path to dataset train directory e.g. /path/to/kinetics/train')
    parser.add_argument('--num_frames', type=int, default=30,
                        help='number of frames to sample per clip')
    parser.add_argument('--target_fps', type=int, default=30,
                        help='video fps')
    parser.add_argument('--sample_rate', type=int, default=1,
                        help='rate to sample frames')
    parser.add_argument('--num_train_clips', type=int, default=1,
                        help='number of clips to sample per videos')
    parser.add_argument('--train_crop_size', type=int, default=112,
                        help="train crop size")
    parser.add_argument('--test_crop_size', type=int, default=112,
                        help="test crop size")
    parser.add_argument('--colorjitter', type='bool', default='True',
                        help='use color jitter')
    parser.add_argument('--use_grayscale', type='bool', default='True',
                        help='use grayscale augmentation')
    parser.add_argument('--use_gaussian', type='bool', default='True',
                        help='use gaussian augmentation')
    parser.add_argument('--num_sec_aud', type=int, default=1,
                        help='number of seconds of audio')
    parser.add_argument('--num_sec', type=int, default=1,
                        help='number of seconds for video, should equal num_frames/fps and num_sec_aud')
    parser.add_argument('--aud_sample_rate', type=int, default=24000,
                        help='audio sample rate')
    parser.add_argument('--audio_augtype', type=str, default='none',
                        choices=['none', 'mild', 'medium', 'heavy'], 
                        help='audio augmentation strength with Spec Augment')
    parser.add_argument('--aud_spec_type', type=int, default=2,
                        help="audio spec type")
    parser.add_argument('--use_volume_jittering', type='bool', default='True',
                        help='use volume jittering')
    parser.add_argument('--use_audio_temp_jittering', type='bool', default='False',
                        help='use audio temporal jittering')
    parser.add_argument('--z_normalize', type='bool', default='False',
                        help='z-normalize the audio')
    parser.add_argument('--dual_data', type='bool', default='False',
                        help='sample two clips per video')

    #########################
    #### optim parameters ###
    #########################
    parser.add_argument('--epochs', default=150, type=int,
                        help='number of total epochs to run')
    parser.add_argument('--batch_size', default=16, type=int,
                        help='batch size per gpu, i.e. how many unique instances per gpu')
    parser.add_argument('--base_lr', default=1e-2, type=float, 
                        help='base learning rate')
    parser.add_argument('--temp', default=0.5, type=float, 
                        help='within-modal NCE temp')
    parser.add_argument('--final_lr', type=float, default=0, 
                        help='final learning rate')
    parser.add_argument('--wd', default=1e-5, type=float, 
                        help='weight decay')
    parser.add_argument('--warmup_epochs', default=10, type=int, 
                        help='number of warmup epochs')
    parser.add_argument('--start_warmup', default=1e-3, type=float,
                        help='initial warmup learning rate')
    parser.add_argument('--lr_milestones', default='20,30,40', type=str, 
                        help='decrease lr on milestones')
    parser.add_argument('--lr_gamma', default=0.1, type=float, 
                        help='decrease lr by a factor of lr-gamma')
    parser.add_argument('--use_lars', default='False', type='bool',
                        help="use LARS optimizer")
    parser.add_argument('--use_warmup_scheduler', default='True', type='bool',
                        help="use warmup scheduler")
    parser.add_argument('--use_lr_scheduler' , default='False', type='bool',
                        help='use cosine LR scheduler')
    parser.add_argument('--multi_crop', type='bool', default='True',
                        help='do multi-crop comparisons')
    parser.add_argument('--use_random_resize_crop', type='bool', default='True',
                        help='use random resized crop instead of short stide jitter')
    parser.add_argument('--cross_modal_alpha', type=float, default=0.5,
                        help='weighting of cross-modal loss')
    parser.add_argument('--num_large_crops', type=int, default=2,
                        help='Number of Large Crops')
    parser.add_argument('--num_small_crops', type=int, default=0,
                        help='Number of small Crops')
    parser.add_argument('--num_large_tcrops', type=int, default=0,
                        help='Number of Large temporal Crops ')
    parser.add_argument('--num_small_tcrops', type=int, default=0,
                        help='Number of small temporal Crops')

    #########################
    #### dist parameters ###
    #########################
    parser.add_argument('--dist_url', default='env://', type=str, 
                        help="""url used to set up distributed