I have a pre-trained NN model which was trained on GPU and now I want to demonstrate some result but I need to do that using CPU (because of resource limitation). I tried to load model states using CPU but getting UNKNOWN error. Everything works perfectly if I use GPU.
Not sure if this information is important - I have used data parallel while training in GPU.