From 9c2a4b232c84dda5122f2763b7f35e355c5862dd Mon Sep 17 00:00:00 2001 From: Pavol Mulinka Date: Thu, 2 Dec 2021 15:45:24 +0000 Subject: [PATCH] LDS tested - working --- pytorch_widedeep/training/_wd_dataset.py | 2 -- pytorch_widedeep/training/trainer.py | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/pytorch_widedeep/training/_wd_dataset.py b/pytorch_widedeep/training/_wd_dataset.py index 48521c1..36456e7 100644 --- a/pytorch_widedeep/training/_wd_dataset.py +++ b/pytorch_widedeep/training/_wd_dataset.py @@ -73,7 +73,6 @@ class WideDeepDataset(Dataset): ] else: self.transforms_names = [] - self.weights = self._prepare_weights(reweight=reweight, lds=lds, lds_kernel=lds_kernel, lds_ks=lds_ks, lds_sigma=lds_sigma) self.Y = target if self.Y is not None: if Ymax is None: @@ -124,7 +123,6 @@ class WideDeepDataset(Dataset): # fill the Bunch X.deepimage = xdi if self.Y is not None: - weight = np.asarray([self.weights[idx]]).astype("float32") if self.weights is not None else self.weights y = self.Y[idx] if self.reweight != None: weight = np.asarray([self.weights[idx]]).astype("float32") diff --git a/pytorch_widedeep/training/trainer.py b/pytorch_widedeep/training/trainer.py index 57404ee..6ec9251 100644 --- a/pytorch_widedeep/training/trainer.py +++ b/pytorch_widedeep/training/trainer.py @@ -636,7 +636,7 @@ class Trainer: self.callback_container.on_eval_begin() self.valid_running_loss = 0.0 with trange(eval_steps, disable=self.verbose != 1) as v: - for i, (data, targett, weight) in zip(v, eval_loader): + for i, (data, targett) in zip(v, eval_loader): v.set_description("valid") val_score, val_loss = self._eval_step(data, targett, i) print_loss_and_metric(v, val_loss, val_score) -- GitLab