Deep Learning, Pytorch, Fine Tuning, Computer Vision, Transfer Learning.
initially in training mode by default after loading a pre-trained model. However, we can switch the model to validation mode during inference or validation. These mode changes alter the model’s behaviour accordingly. Training mode When the model is in training mode, it enables specific operations