Hi, thank you for reply!
Yes, I actually had a global optimizer under the “opt” variable. Changing from
train_stats = train(model, opt, train_loader)

to
train_stats = train(model, optimizer, train_loader)

in train_model() fixed the issue.