Hi, it’s a lot of code so can’t help you with everything. But, you should replace model.forward(images) with model(images).
Common advice is to try to overfit to a small training set to figure out if the code can actually learn something