My training loader’s length is 214. I actually solved that problem a different way, but I have now run into another problem of the wave decrease at a small rate.
The change I was made was relocating lr.step():
for i in range(epoch):
        trn_corr = 0
        tst_corr = 0
         #Run training for one epoch
        train(train_loader, MobileNet, criterion, optimizer, i,trn_corr)
        #evaluate the validation/test
        test(test_loader, MobileNet, criterion, i, epoch,tst_corr)
        lrs.step()