Thank you so much for your response. However when I tried this code, i got this error
TypeError: can't assign a numpy.ndarray to a torch.cuda.FloatTensor
Please can you help me in regard to this ? Any help is much appreciated. Thanks in advance