From a500b3545f6eb25c480e945f2e12786933c92423 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Wed, 21 Jun 2017 08:35:16 +0200 Subject: [PATCH] Update. --- svrtset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/svrtset.py b/svrtset.py index 0a14121..ecbaa68 100755 --- a/svrtset.py +++ b/svrtset.py @@ -114,8 +114,8 @@ class CompressedVignetteSet: for b in range(0, self.nb_batches): target = torch.LongTensor(self.batch_size).bernoulli_(0.5) input = svrt.generate_vignettes(problem_number, target) - acc += input.sum() / input.numel() - acc_sq += (input * input).sum() / input.numel() + acc += float(input.sum()) / input.numel() + acc_sq += float((input * input).sum()) / input.numel() self.targets.append(target) self.input_storages.append(svrt.compress(input.storage())) if logger is not None: logger(self.nb_batches * self.batch_size, b * self.batch_size) -- 2.20.1