MNIST error

HaoTieu · October 8, 2021, 11:06am

I’m studying DL subject at my college.
I want to use fastai to complete my exercise, but I get an error.
Please help me to fix it.

github.com

haotieu2001/haotieu.github.io/blob/master/19521480_BaiThucHanh1.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "19521480_BaiThucHanh1.ipynb",
      "provenance": [],
      "collapsed_sections": [],
      "authorship_tag": "ABX9TyPgn8EIU/MCCpMHbrrvQZcQ",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU"
  },

This file has been truncated. show original

Conwyn · October 8, 2021, 6:44pm

Hao
Try switching to the CPU rather than the GPU to see if the problem disappears. You can also try Googling the message and see whether the dimension mis-match fits your problem.
Regards Conwyn

HaoTieu · October 9, 2021, 3:31am

I use CELoss and I don’t get errors. Why my teacher use BCELoss for MNIST?