mathematics:

46656


I follow your orientations and I had another problem:
Using GPU: True

Epoch 1/10
----------
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-102-cd4c03781ec7> in <module>()
      8 exp_lr_scheduler = lr_scheduler.StepLR(optimizer, step_size=5, gamma=0.1) #decay
      9                                                             # 0.1 every 5 epochs
---> 10 model_ft = train_model(net, criterion, optimizer, exp_lr_scheduler, num_epochs=10)

5 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/functional.py in linear(input, weight, bias)
   1672     if input.dim() == 2 and bias is not None:
   1673         # fused op is marginally faster
-> 1674         ret = torch.addmm(bias, input, weight.t())
   1675     else:
   1676         output = input.matmul(weight.t())

RuntimeError: mat1 dim 1 must match mat2 dim 0