From af8862d360557819f1ff296d85a175835482ad09 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Matt=C3=A9o=20Delabre?= Date: Wed, 18 Dec 2019 11:01:45 -0500 Subject: [PATCH] Reflect parameter changes in `experiment.py` --- experiment.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/experiment.py b/experiment.py index 511c989..8816b05 100644 --- a/experiment.py +++ b/experiment.py @@ -35,9 +35,8 @@ def train_with_parameters( """ model = constellation.ConstellationNet( order=order, - encoder_layers_sizes=layer_sizes, - decoder_layers_sizes=layer_sizes[::-1], - channel_model=constellation.GaussianChannel() + encoder_layers=layer_sizes, + decoder_layers=layer_sizes[::-1], ) # List of training examples (not shuffled)