When the model is in model.train(), and if we just keep passing image through model
model.train()
for i, data in enumerate(data_loader):
x,y = data
f = model(x)
Is question the model learning high activation features?