This commit is contained in:
Dr. Kashif Rasul
2020-05-25 12:32:31 +02:00
parent 674929344b
commit c21a353d5c
+3 -3
View File
@@ -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
}
}