error in training stage "Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
Hello guys!
we are running the recipe of ljspeech of version 0.8.0 but on our data set, we are using google colaboratory. At the stage of training we are getting this error "Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!". Any help ? 
log file is attached.
[train.log](https://github.com/espnet/espnet/files/5044468/train.log)

Thanks in advance
diff --git a/espnet/nets/pytorch_backend/e2e_tts_tacotron2.py b/espnet/nets/pytorch_backend/e2e_tts_tacotron2.py
index 90e28eb59..6c0c64203 100644
--- a/espnet/nets/pytorch_backend/e2e_tts_tacotron2.py
+++ b/espnet/nets/pytorch_backend/e2e_tts_tacotron2.py
@@ -122,7 +122,7 @@ class GuidedAttentionLoss(torch.nn.Module):
 
         """
         grid_x, grid_y = torch.meshgrid(torch.arange(olen), torch.arange(ilen))
-        grid_x, grid_y = grid_x.float(), grid_y.float()
+        grid_x, grid_y = grid_x.float().to(olen.device), grid_y.float().to(ilen.device)
         return 1.0 - torch.exp(
             -((grid_y / ilen - grid_x / olen) ** 2) / (2 * (sigma ** 2))
         )
['espnet/nets/pytorch_backend/e2e_tts_tacotron2.py']