did you try this

  
      github.com
  
  
    patil-suraj/Notebooks/blob/master/longformer_qa_training.ipynb 8
{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "longformer-qa-training",
      "provenance": [],
      "collapsed_sections": [
        "tty8vuMBqI5L"
      ],
      "machine_shape": "hm",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "accelerator": "GPU",
    "widgets": {
      "application/vnd.jupyter.widget-state+json": {


  This file has been truncated. show original