fde2f66676f960782c993f7148927c4a4197ab10,spotlight/factorization/explicit.py,ExplicitFactorizationModel,fit,#ExplicitFactorizationModel#Any#Any#,85

Before Change

                self._use_cuda
            )
        else:
            self._net = gpu(
                TruncatedBilinearNet(self._num_users,
                                     self._num_items,
                                     self._embedding_dim,
                                     sparse=self._sparse),
                self._use_cuda
            )

        if self._optimizer is None:
            self._optimizer = optim.Adam(self._net.parameters())

After Change

                                  self._use_cuda)
            item_ids_tensor = gpu(torch.from_numpy(items),
                                  self._use_cuda)
            ratings_tensor = gpu(torch.from_numpy(ratings),
                                 self._use_cuda)

            epoch_loss = 0.0

            for (batch_user,

Instances