Colab pro -fine RoBERTa error tcmalloc: large alloc 6325288960
I want to fine-tune RoBERTa on my newspaper data  (around 8GB) using Colab pro. It works fine on small data. I have given my codes below. Are my codes correct? **Is there any way to handle this memory problem**? The error crushes the colab.

tcmalloc: large alloc 6325288960 bytes == 0x447fa000 @  0x7f3438dcc1e7 0x59221c 0x4ca6f4 0x566daa 0x5a4df1 0x5a5eea 0x4ce082 0x566c02 0x5a4df1 0x5a60ae 0x5bd138 0x50a47f 0x50c1f4 0x507f24 0x509202 0x594b01 0x54a17f 0x5517c1 0x5a9eec 0x50a783 0x50cfd6 0x507f24 0x509c50 0x50a64d 0x50cfd6 0x509918 

 ```

!python "/content/transformers/examples/language-modeling/run_language_modeling.py" \
    --output_dir "/content/drive/My Drive/EPU-NLP/FinetuneModel/output" \
    --model_type roberta \
    --model_name_or_path roberta-base \
    --do_train \
    --per_gpu_train_batch_size 16 \
    --seed 22 \
    --train_data_file "/content/drive/My Drive/EPU-NLP/FinetuneModel/data_all.txt" \
    --block_size 256 \
    --line_by_line \
    --weight_decay 0.01 \
    --adam_epsilon 1e-6 \
    --save_total_limit 500 \
    --learning_rate 6e-4 \
    --num_train_epochs 3 \
    --save_total_limit 500 \
    --save_steps 500 \
    --mlm
```

Got the same with StyleGAN using Colab Pro<|||||>What is your data size? First, try it with 2GB of data.  Also check again by decreasing the GPU batch size to 8 or 4.<|||||>This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.
<|||||>This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the [contributing guidelines](https://github.com/huggingface/transformers/blob/master/CONTRIBUTING.md) are likely to be ignored.