The error explains, that your data is pushed to the GPU (torch.cuda.FloatTensor), while the parameters of your model are still on the CPU.
To push all parameters and buffers of your model to the GPU, you could use:
model.to('cuda')