my code looks like this :

    parser = argparse.ArgumentParser()
    parser.add_argument('-c', '--config-file', type=str, default='train_config.yaml',
                        help='train config file')
    parser.add_argument('-t', '--train-times', type=int, default=1,
                        help='train the same model several times')
    parser.add_argument('--dataset_dir', help='path to folder containing the preped dataset.', required=True)
    parser.add_argument('--backup', action='store_true', help='store the config file')
       
    parser.add_argument('--device', type=int, default=0,
                        help='cuda device index to run the training')

    task.connect(parser)

    args = parser.parse_args()