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