(base) root@DESKTOP-QS39DMG:/mnt/f/Users/_/Documents/doing/prac/py/mnist-fast-ai# python run.py
epoch     train_loss  valid_loss  accuracy  time
Traceback (most recent call last):
  File "run.py", line 25, in <module>
    main()
  File "run.py", line 23, in main
    learn.fit_one_cycle(1)
  File "/root/anaconda3/lib/python3.7/site-packages/fastai/train.py", line 22, in fit_one_cycle
    learn.fit(cyc_len, max_lr, wd=wd, callbacks=callbacks)
  File "/root/anaconda3/lib/python3.7/site-packages/fastai/basic_train.py", line 199, in fit
    fit(epochs, self, metrics=self.metrics, callbacks=self.callbacks+callbacks)
  File "/root/anaconda3/lib/python3.7/site-packages/fastai/basic_train.py", line 101, in fit
    loss = loss_batch(learn.model, xb, yb, learn.loss_func, learn.opt, cb_handler)
  File "/root/anaconda3/lib/python3.7/site-packages/fastai/basic_train.py", line 26, in loss_batch
    out = model(*xb)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "run.py", line 12, in forward
    x = self.lin1(xb)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/linear.py", line 92, in forward
    return F.linear(input, self.weight, self.bias)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/functional.py", line 1408, in linear
    output = input.matmul(weight.t())
RuntimeError: size mismatch, m1: [2688 x 28], m2: [784 x 50] at /opt/conda/conda-bld/pytorch_1556653114079/work/aten/src/TH/generic/THTensorMath.cpp:961