get_args.py

__author__ = 'kirtyvedula'

import argparse

def get_args():
    ################################
    # Setup all user parameters and get args
    ################################
    parser = argparse.ArgumentParser()
    parser.add_argument('-n_channel', type=float, default = 7)
    parser.add_argument('-k', type=float, default = 4)
    parser.add_argument('-bec_p',type=float, default=0.0, help ='only for bec channel')
    parser.add_argument('--no-cuda', action='store_true', default=False, help='disables CUDA training')
    parser.add_argument('-modulation', choices = ['BPSK','QPSK','8PSK','16QAM','64QAM','256QAM'], default='BPSK')
    parser.add_argument('-coding', choices=['SingleParity_4_3','Hamming_7_4','Hamming_15_11','Polar_16_4', 'EGolay_24_12'],default='Hamming_7_4')
    parser.add_argument('-learning_rate', type=float, default=0.001)
    parser.add_argument('-batch_size', type=int, default=64)
    parser.add_argument('-dropout',type=float, default=0.0)
    parser.add_argument('-EbN0_dB_train', type=float, default=3.0)
    parser.add_argument('-EbN0dB_test_start', type=float, default=-0.0)
    parser.add_argument('-EbN0dB_test_end', type=float, default=11.5)
    parser.add_argument('-EbN0dB_precision', type=int, default=0.5)
    parser.add_argument('-epochs', type=int, default=10)
    args = parser.parse_args()

    return args