From 826b93f6b5703d1c0a555eb6e670d02aad263a7b Mon Sep 17 00:00:00 2001 From: Claudio Scheer Date: Mon, 6 Apr 2020 01:32:17 -0300 Subject: [PATCH] Change the number of hidden layer --- mpl-xor/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mpl-xor/main.py b/mpl-xor/main.py index 99ccf65..82a2886 100644 --- a/mpl-xor/main.py +++ b/mpl-xor/main.py @@ -16,7 +16,7 @@ class MultilayerPerceptron(nn.Module): return self.layers(x) -model = MultilayerPerceptron(5) +model = MultilayerPerceptron(3) model.cuda() loss_function = nn.MSELoss() -- GitLab