Validation loss and training loss is not increasing in pytorch
you can check if you can overfit by training on a much smaller dataset. For example, subset your dataset to only 10 images. In this dataset, you should be able to get low training loss, if