Unable to load DistilBertModel during training 
Author: BatMrE
Pull request: no

I'm following the example to train a DistilBert model from scratch from: examples/distillation/README.md

[https://github.com/huggingface/transformers/blob/1ab8dc44b3d84ed1894f5b6a6fab58fb39298fc7/examples/distillation/README.md](url)

during preparation of data converted my json training data to text and generated binarized_text using script:

```
 python scripts/binarized_data.py \
     --file_path data/dump.txt \
     --tokenizer_type bert \
     --tokenizer_name bert-base-uncased \
     --dump_file data/binarized_text

```
After that I got pickle file generated in data folder, created **serialization_dir/my_first_training** folder.
now when I have all the pre processed data and trying to train to get distilbert model using script:

```
 python train.py \
     --student_type distilbert \
     --student_config training_configs/distilbert-base-uncased.json \
     --teacher_type bert \
     --teacher_name bert-base-uncased \
     --alpha_ce 5.0 --alpha_mlm 2.0 --alpha_cos 1.0 --alpha_clm 0.0 --mlm \
     --freeze_pos_embs \
     -- n_gpu 1 \
     --dump_path serialization_dir/my_first_training \
     --data_file data/binarized_text.bert-base-uncased.pickle \
     --token_counts data/token_counts.bert-base-uncased.pickle \
     --force # overwrites the `dump_path` if it already exists.
 
```

I am getting following error even thoe I have **dump_path, data_file**:
`train.py: error: the following arguments are required: --dump_path, --data_file`

full stack:
```
2021-04-28 11:55:58.684284: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory
2021-04-28 11:55:58.684326: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
usage: train.py [-h] [--force] --dump_path DUMP_PATH --data_file DATA_FILE
                --student_type {distilbert,roberta,gpt2} --student_config
                STUDENT_CONFIG
                [--student_pretrained_weights STUDENT_PRETRAINED_WEIGHTS]
                --teacher_type {bert,roberta,gpt2} --teacher_name TEACHER_NAME
                [--temperature TEMPERATURE] [--alpha_ce ALPHA_CE]
                [--alpha_mlm ALPHA_MLM] [--alpha_clm ALPHA_CLM]
                [--alpha_mse ALPHA_MSE] [--alpha_cos ALPHA_COS] [--mlm]
                [--mlm_mask_prop MLM_MASK_PROP] [--word_mask WORD_MASK]
                [--word_keep WORD_KEEP] [--word_rand WORD_RAND]
                [--mlm_smoothing MLM_SMOOTHING] [--token_counts TOKEN_COUNTS]
                [--restrict_ce_to_mask] [--freeze_pos_embs]
                [--freeze_token_type_embds] [--n_epoch N_EPOCH]
                [--batch_size BATCH_SIZE] [--group_by_size]
                [--gradient_accumulation_steps GRADIENT_ACCUMULATION_STEPS]
                [--warmup_prop WARMUP_PROP] [--weight_decay WEIGHT_DECAY]
                [--learning_rate LEARNING_RATE] [--adam_epsilon ADAM_EPSILON]
                [--max_grad_norm MAX_GRAD_NORM]
                [--initializer_range INITIALIZER_RANGE] [--fp16]
                [--fp16_opt_level FP16_OPT_LEVEL] [--n_gpu N_GPU]
                [--local_rank LOCAL_RANK] [--seed SEED]
                [--log_interval LOG_INTERVAL]
                [--checkpoint_interval CHECKPOINT_INTERVAL]
train.py: error: the following arguments are required: --dump_path, --data_file
```

@patil-suraj 

Comments:
Does removing the space between `--` and `n_gpu` help?

Yes @LysandreJik I was trying to use gpu device 1 , later made device changes from code .Thanks