Issue with prediction method #32
Description
OS: ubuntu 16.04
GPU: Nvidia GeForce GTX 105
The kernel dies while loading network when working from docker
and when I set up my own environment I am able to run it through some layers only from net.forward() method but unable to get to the output, the kernel dies on forward pass. kindly suggest how to solve the issue as I am testing it only for one image right now as given in the notebook. Thank you
