Weather Data Image Classification notebook shows "IndexError: Target 4 is out of bounds"

[typex1]

Link to the notebook
Weather Data Image Classification notebook

What aspects of the notebook do not work?

For the following code cell in section "5. Fine-tune the pre-trained model on the labelled weather data"
print('Training of Model:') model_ft = train_model(model_ft, criterion, optimizer_ft,model_checkpoint=0,num_epochs=epochs)
I get the following error:

`---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
Cell In[26], line 2
1 print('Training of Model:')
----> 2 model_ft = train_model(model_ft, criterion, optimizer_ft,model_checkpoint=0,num_epochs=epochs)

Cell In[12], line 35, in train_model(model, criterion, optimizer, model_checkpoint, early_stop, num_epochs)
33 _, preds = torch.max(output, 1)
34 # calculate the batch loss
---> 35 loss = criterion(output, target)
36 # backward pass: compute gradient of the loss with respect to model parameters
37 loss.backward()

File ~/.conda/envs/cv/lib/python3.9/site-packages/torch/nn/modules/module.py:1194, in Module._call_impl(self, *input, **kwargs)
1190 # If we don't have any hooks, we want to skip the rest of the logic in
1191 # this function, and just call forward.
1192 if not (self._backward_hooks or self._forward_hooks or self._forward_pre_hooks or _global_backward_hooks
1193 or _global_forward_hooks or _global_forward_pre_hooks):
-> 1194 return forward_call(*input, **kwargs)
1195 # Do not call functions when jit is used
1196 full_backward_hooks, non_full_backward_hooks = [], []

File ~/.conda/envs/cv/lib/python3.9/site-packages/torch/nn/modules/loss.py:1174, in CrossEntropyLoss.forward(self, input, target)
1173 def forward(self, input: Tensor, target: Tensor) -> Tensor:
-> 1174 return F.cross_entropy(input, target, weight=self.weight,
1175 ignore_index=self.ignore_index, reduction=self.reduction,
1176 label_smoothing=self.label_smoothing)

File ~/.conda/envs/cv/lib/python3.9/site-packages/torch/nn/functional.py:3026, in cross_entropy(input, target, weight, size_average, ignore_index, reduce, reduction, label_smoothing)
3024 if size_average is not None or reduce is not None:
3025 reduction = _Reduction.legacy_get_string(size_average, reduce)
-> 3026 return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)

IndexError: Target 4 is out of bounds.`

[MicheleMonclova]

This looks like a Pytorch issue that may be in the code.
did you see this: https://discuss.pytorch.org/t/indexerror-target-4-is-out-of-bounds/99172