This is my model description. Should I change
model.fc = nn.Sequential(nn.Linear(2048, 512),
                         nn.ReLU(),
                         nn.Dropout(0.2),
                         nn.Linear(512, 10),
                         nn.LogSoftmax(dim=1))

to
model.fc = nn.Sequential(nn.Linear(2048, 512),
                         nn.ReLU(),
                         nn.Dropout(0.2),
                         nn.Linear(512, 225),
                         nn.LogSoftmax(dim=1))

Edit:
This creates the same error. Now Target 225 is out of bounds.
Could this happen because I do not specify targets while I’m loading the images?