From c21a353d5cb627dcbfa95122fdf4f19855b150ab Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Mon, 25 May 2020 12:32:31 +0200 Subject: [PATCH] fix typo --- examples/Multivariate-Flow-Solar.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/Multivariate-Flow-Solar.ipynb b/examples/Multivariate-Flow-Solar.ipynb index c69ce49..88a04be 100644 --- a/examples/Multivariate-Flow-Solar.ipynb +++ b/examples/Multivariate-Flow-Solar.ipynb @@ -33,7 +33,7 @@ "metadata": {}, "outputs": [], "source": [ - "device = torch.device(\"cuda:3\" if torch.cuda.is_available() else \"cpu\")" + "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")" ] }, { @@ -49,7 +49,7 @@ "metadata": {}, "outputs": [], "source": [ - "dataset = get_dataset(\"solar_nips\", regenerate=False)" + "dataset = get_dataset(\"solar_nips\", regenerate=False, shuffle=False)" ] }, { @@ -1399,4 +1399,4 @@ }, "nbformat": 4, "nbformat_minor": 4 -} +} \ No newline at end of file