diff --git a/omniglot/omniglot_train_few_shot.py b/omniglot/omniglot_train_few_shot.py
index 58e0b529..337b048c 100644
--- a/omniglot/omniglot_train_few_shot.py
+++ b/omniglot/omniglot_train_few_shot.py
@@ -203,7 +203,7 @@ def main():
         relation_network_optim.step()
 
         if (episode+1)%100 == 0:
-                print("episode:",episode+1,"loss",loss.data[0])
+                print("episode:",episode+1,"loss",loss.item())
 
         if (episode+1)%5000 == 0: