So I made model to do Super Resolution over MNIST dataset, I’ve already trained a classifier to whether or not it can recognize the number from the output of my SR model.
I use the results of my classifier to comput the loss of my SR model and the question I have is this one :
During the training loop
SRmodel.train()
classifier.eval()
ouptut = SRmodel(data)
predictedLabel = classifier(output)
loss = MyLoss(data, output, predictedLabel, label)
loss.backward()
How do I know whether the backward propagation is occurring only over my SR model, or over my classifier, or both? (So far the output of my SR model are correct, but I might be missing something)