def parse_args(parser):
    parser.add_argument('--dataset', default='pawsx', type=str, help='pawsx')
    parser.add_argument('--bs', default=32, type=int, help='batch size')
    parser.add_argument('--epochs', default=100, type=int, help='training epochs')
    parser.add_argument('--dataset_path', default='/', type=str, help='dataset path')