I am training model on CIFAR-10 dataset and after every epoch I evaluate its testing accuracy with current best testing accuracy. Once the testing accuracy is higher than the best one I save the both the model and optimizer. After that I resume training of the model using the saved one.
Now whenever I save the best model I wanted to visualize the response map of a layer using the same(like using ‘./model_checkpoint.pth’). What is the approach of doing this task.