actually i am trying to tune the hyperparameters by looping the code by changing learning rate, batch size and number of epochs
the loss appears to be oscilating for all above values around the initial value
also i tried to change weight decay and momentum