Getting error while running predict module

[kiran-ray]

Getting this error
RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 'weight'

Steps:
Executed train.py in GPU mode.
Similarly executed predict.py in GPU mode. Looks like the GPU mode settings are not done in predict flow?