Reflect parameter changes in `experiment.py`

2019-12-18 11:01:45 -05:00 · 2019-12-18 11:01:45 -05:00 · af8862d360
parent d8a140d793
commit af8862d360
1 changed files with 2 additions and 3 deletions
--- 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)