Thank you. It worked and I also got accuracy of 97.46%
Thank you again for your timely and needy help.
I need one more help from you. Please let me know how this program can be modified to work with making use of CUDA Graph. This is the last thing I need for my thesis purpose.
I have written sample program with CUDA Graph but unable to implement it for this program.
So I request you to please provide a code how to convert this network_api_pytorch_mnist code to make use of CUDA Graph.