Solved by dhpollack in post #2 
              
              
                Your inputs are float64 (Double) tensors, but the network expects a float32 tensor.  You should either convert the tensor to float with .float() before putting it into the network.  Also, you haven’t really given us enough information about your setup but this is probably because you are creating yo…