def parse_args():
    parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('--num_proc', type=int, default=1, help='dataset process thread num')
    parser.add_argument('--max_length', type=int, default=64, help='sentence max length')
    parser.add_argument('--batch_size', type=int, default=32, help='batch size')
    parser.add_argument('--epochs', type=int, default=101, help='epochs')
    parser.add_argument('--lr', type=float, default=1e-05, help='learning rate')
    parser.add_argument('--tao', type=float, default=0.05, help='temperature')
    parser.add_argument('--device', type=str, default='cuda', help='device')
    parser.add_argument('--display_interval', type=int, default=500, help='display interval')
    parser.add_argument('--save_interval', type=int, default=10, help='save interval')
    parser.add_argument('--pool_type', type=str, default='pooler', help='pool_type')
    parser.add_argument('--dropout_rate', type=float, default=0.3, help='dropout_rate')
    args = parser.parse_args()
    return args
def parse_args():
    parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('--num_proc', type=int, default=1, help='dataset process thread num')
    parser.add_argument('--max_length', type=int, default=64, help='sentence max length')
    parser.add_argument('--batch_size', type=int, default=32, help='batch size')
    parser.add_argument('--epochs', type=int, default=101, help='epochs')
    parser.add_argument('--lr', type=float, default=1e-05, help='learning rate')
    parser.add_argument('--tao', type=float, default=0.05, help='temperature')
    parser.add_argument('--device', type=str, default='cuda', help='device')
    parser.add_argument('--display_interval', type=int, default=500, help='display interval')
    parser.add_argument('--save_interval', type=int, default=10, help='save interval')
    parser.add_argument('--pool_type', type=str, default='pooler', help='pool_type')
    parser.add_argument('--dropout_rate', type=float, default=0.3, help='dropout_rate')
    args = parser.parse_args()
    return args