diff --git a/zounds/learn/wgan.py b/zounds/learn/wgan.py
index <HASH>..<HASH> 100644
--- a/zounds/learn/wgan.py
+++ b/zounds/learn/wgan.py
@@ -230,9 +230,9 @@ class WassersteinGanTrainer(Trainer):
                 g_loss.backward()
                 self.generator_optim.step()
 
-                gl = g_loss.data[0]
-                dl = d_loss.data[0]
-                rl = real_mean.data[0]
+                gl = g_loss.data.item()
+                dl = d_loss.data.item()
+                rl = real_mean.data.item()
 
                 self.on_batch_complete(
                     epoch=epoch,
Use new item() method after upgrade to pytorch <I>