{
  "id": 211902,
  "title": "How to utilize TPU with PytorchLightning?",
  "url": "/competitions/cassava-leaf-disease-classification/discussion/211902",
  "author_name": "jabertuhin",
  "post_date": "2021-01-16T18:58:15.515000",
  "votes": 1,
  "comment_count": 0,
  "views": 0,
  "content": "<p>I have been trying to use TPU for this competition via PytorchLightning Trainer.<br>\nIt's not using TPU properly, with large batch size - CPU is getting exhausted and RAM usage crossing allowed limit. </p>\n<p>Here is my notebook - <a href=\"https://www.kaggle.com/jabertuhin/classification-with-pytorchlightning-using-tpu\" target=\"_blank\">Classification with PytorchLightning, using TPU</a></p>\n<p>It was taking more than 3 hours to complete 5 epoch training. </p>\n<p>Any suggestion?</p>",
  "messages": [
    {
      "id": 1155902,
      "postDate": "2021-01-16T18:58:15.517Z",
      "content": "<p>I have been trying to use TPU for this competition via PytorchLightning Trainer.<br>\nIt's not using TPU properly, with large batch size - CPU is getting exhausted and RAM usage crossing allowed limit. </p>\n<p>Here is my notebook - <a href=\"https://www.kaggle.com/jabertuhin/classification-with-pytorchlightning-using-tpu\" target=\"_blank\">Classification with PytorchLightning, using TPU</a></p>\n<p>It was taking more than 3 hours to complete 5 epoch training. </p>\n<p>Any suggestion?</p>",
      "rawMarkdown": "I have been trying to use TPU for this competition via PytorchLightning Trainer.\nIt's not using TPU properly, with large batch size - CPU is getting exhausted and RAM usage crossing allowed limit. \n\nHere is my notebook - [Classification with PytorchLightning, using TPU](https://www.kaggle.com/jabertuhin/classification-with-pytorchlightning-using-tpu)\n\nIt was taking more than 3 hours to complete 5 epoch training. \n\nAny suggestion?",
      "votes": 1
    }
  ],
  "comments": [],
  "raw_markdown_by_id": {
    "1155902": "I have been trying to use TPU for this competition via PytorchLightning Trainer.\nIt's not using TPU properly, with large batch size - CPU is getting exhausted and RAM usage crossing allowed limit. \n\nHere is my notebook - [Classification with PytorchLightning, using TPU](https://www.kaggle.com/jabertuhin/classification-with-pytorchlightning-using-tpu)\n\nIt was taking more than 3 hours to complete 5 epoch training. \n\nAny suggestion?"
  }
}