def partial_to(self, device):
        """Put parameters into device except W_r

        Parameters
        ----------
        device : torch device
            Which device the parameters are put in.
        """
        assert device == self.device
        if device is not None:
            self.ufc.cuda(device)
            if self.share_user_item_param is False:
                self.ifc.cuda(device)
            self.dropout.cuda(device)
Put parameters into device except W_r

        Parameters
        ----------
        device : torch device
            Which device the parameters are put in.
        
examples/pytorch/gcmc/model.py