Fine tuning for image classification using Pytorch

Pytorch does not come with "fit" and "predict" functions. We need a training loop! Now for training, we need the dataset loaders from torch, an optimizer and a learning rate scheduler. Please note that we have defined the loss criterion inside the training function. The loss function we are using here is Binary Cross Entropy with Logits (BCEWithLogitsLoss).