Yes, this part of code 41 transforms image size to 224*224 explicitly. Thanks for telling, I’ve changed to 299*299.
Unfortunately, I am getting the following error now -
Traceback (most recent call last):
  File "main.py", line 316, in <module>
    main()
  File "main.py", line 158, in main
    train(train_loader, model, criterion, optimizer, epoch)
  File "main.py", line 196, in train
    loss = criterion(output, target_var)
  File "/home/abhishs8/Research/Misc_Experiments/Mystuff/.torch-env/lib/python3.6/site-packages/torch/nn/modules/module.py", line 325, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/abhishs8/Research/Misc_Experiments/Mystuff/.torch-env/lib/python3.6/site-packages/torch/nn/modules/loss.py", line 601, in forward
    self.ignore_index, self.reduce)
  File "/home/abhishs8/Research/Misc_Experiments/Mystuff/.torch-env/lib/python3.6/site-packages/torch/nn/functional.py", line 1140, in cross_entropy
    return nll_loss(log_softmax(input, 1), target, weight, size_average, ignore_index, reduce)
  File "/home/abhishs8/Research/Misc_Experiments/Mystuff/.torch-env/lib/python3.6/site-packages/torch/nn/functional.py", line 786, in log_softmax
    return torch._C._nn.log_softmax(input, dim)
RuntimeError: log_softmax(): argument 'input' (position 1) must be Variable, not tuple