Sorry. Yeah, that makes more sense. But I still end up with values like 27855.0 for train_epoch_acc/len(train_loader)
train_epoch_acc/len(train_loader)
