I am getting the following error of doing a forward pass on my model
  File "/media/newDrive/rishabh/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in __call__
    result = self.forward(*input, **kwargs)
  File "/media/newDrive/rishabh/anaconda3/lib/python3.6/site-packages/torch/nn/modules/batchnorm.py", line 49, in forward
    self.training or not self.track_running_stats, self.momentum, self.eps)
  File "/media/newDrive/rishabh/anaconda3/lib/python3.6/site-packages/torch/nn/functional.py", line 1194, in batch_norm
    training, momentum, eps, torch.backends.cudnn.enabled
RuntimeError: get_device is not implemented for type torch.FloatTensor


Can someone explain how to fix this and why is this happening??