From eebaf4e03bd9278325f4f0d4de2ea70c00ebd030 Mon Sep 17 00:00:00 2001 From: Javier Vargas Date: Mon, 1 Nov 2021 13:59:59 +0100 Subject: [PATCH] experiment training --- experiment1/dataset.py | 2 +- experiment1/modules.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/experiment1/dataset.py b/experiment1/dataset.py index 7d55057..586de3d 100644 --- a/experiment1/dataset.py +++ b/experiment1/dataset.py @@ -7,7 +7,7 @@ class NumberAdd(Dataset): def __init__(self, total_samples: int) -> None: """Generate dataset with pairs of numbers and their sum""" self.N = total_samples - self.samples = T.randint(low=-1000, high=1000, size=(self.N, 2)) + self.samples = T.randint(low=-1000, high=1000, size=(self.N, 2), dtype=T.float32) self.targets = T.sum(self.samples, dim=-1) def __getitem__(self, index: int) -> tuple: diff --git a/experiment1/modules.py b/experiment1/modules.py index 739e3ce..d578953 100644 --- a/experiment1/modules.py +++ b/experiment1/modules.py @@ -8,7 +8,7 @@ def __init__(self) -> None: def __call__(self, X: T.Tensor) -> T.Tensor: """Calling the adding operation without gradient""" - return X[0] + X[1] + return X[:, 0] + X[:, 1] class Substracter(object): @@ -18,7 +18,7 @@ def __init__(self) -> None: def __call__(self, X: T.Tensor) -> T.Tensor: """Calling the adding operation without gradient""" - return X[0] - X[1] + return X[:, 0] - X[:, 1] if __name__ == "__main__":