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??