7d2f2ed761ffc49feb337bbee878e8256af568a8,livelossplot/output_plugins/matplotlib_subplots.py,Plot2d,_predict_pytorch,#Plot2d#,107

Before Change

    def _predict_pytorch(self, model, x_numpy):
        import torch
        x = torch.from_numpy(x_numpy).float()
        return model(x).softmax(dim=1).detach().numpy()

    def predict(self, model, X):
        // e.g. model(torch.fromnumpy(X)).detach().numpy()

After Change

    def _predict_pytorch(self, model, x_numpy):
        import torch
        x = torch.from_numpy(x_numpy).to(self.torch_device).float()
        return model(x).softmax(dim=1).detach().cpu().numpy()

    def predict(self, model, X):
        // e.g. model(torch.fromnumpy(X)).detach().numpy()

Instances