Replace .long() by dtypes

Author: tristandeleu

modules.py

@@ -191,8 +191,8 @@ def forward(self, x, label):
 
     def generate(self, label, shape=(8, 8), batch_size=64):
         param = next(self.parameters())
-        x = torch.zeros(batch_size, *shape).long()
-        x = x.to(param.device)
+        x = torch.zeros((batch_size, *shape),
+            dtype=torch.int64, device=param.device)
 
         for i in range(shape[0]):
             for j in range(shape[1]):

pixelcnn.py

@@ -115,7 +115,7 @@ def test():
 
 def generate_samples():
     label = torch.arange(10).expand(10, 10).contiguous().view(-1)
-    label = label.long().to(DEVICE)
+    label = label.to(device=DEVICE, dtype=torch.int64)
 
     latents = model.generate(label, shape=LATENT_SHAPE, batch_size=100)
     x_tilde, _ = autoencoder.decode(latents)