diff --git a/cornac/models/gcmc/nn_modules.py b/cornac/models/gcmc/nn_modules.py
index 78a26fe7799ccb09daf7d5e4b1fc2aba197fc6ed..fcc74545c2d5a9446067217437e6189be2d8ac5c 100644
--- a/cornac/models/gcmc/nn_modules.py
+++ b/cornac/models/gcmc/nn_modules.py
@@ -62,8 +62,8 @@ class NeuralNetwork(nn.Module):
         ifeat : torch.Tensor
             The input item feature
         """
-        user_out, item_out = self.encoder(enc_graph, ufeat, ifeat)
-        pred_ratings = self.decoder(dec_graph, user_out, item_out)
+        self.user_out, self.item_out = self.encoder(enc_graph, ufeat, ifeat)
+        pred_ratings = self.decoder(dec_graph, self.user_out, self.item_out)
         return pred_ratings