I’ll try right now and let you now in a minute.
This is the lr scheduler:
exp_lr_scheduler = optim.lr_scheduler.StepLR(optimizer, step_size=5, gamma=0.1)

Since it’s a pretrained net and I only need to twitch a little the last layer, I was thinking about training it only for 15 epochs.