fix for issue fastai#60

jakubLangr · Jan 31, 2020 · 2afcfcc · 2afcfcc
1 parent 94d219e
commit 2afcfcc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fastai2/callback/wandb.py b/fastai2/callback/wandb.py
@@ -45,7 +45,7 @@ def begin_fit(self):
 
             items = [self.dls.valid_ds.items[i] for i in idxs]
             test_tls = [tl._new(items, split_idx=1) for tl in self.dls.valid_ds.tls]
-            self.valid_dl = self.dls.valid_dl.new(Datasets(tls=test_tls), bs=self.n_preds)
+            self.valid_dl = DataLoader(self.dls.valid, bs=self.n_preds)
 
     def after_batch(self):
         "Log hyper-parameters and training loss"

diff --git a/nbs/70_callback.wandb.ipynb b/nbs/70_callback.wandb.ipynb
@@ -105,7 +105,7 @@
     "\n",
     "            items = [self.dls.valid_ds.items[i] for i in idxs]\n",
     "            test_tls = [tl._new(items, split_idx=1) for tl in self.dls.valid_ds.tls]\n",
-    "            self.valid_dl = self.dls.valid_dl.new(Datasets(tls=test_tls), bs=self.n_preds)\n",
+    "            self.valid_dl = DataLoader(self.dls.valid, bs=self.n_preds)\n",
     "\n",
     "    def after_batch(self):\n",
     "        \"Log hyper-parameters and training loss\"\n",