Reflect parameter changes in `experiment.py`

This commit is contained in:
Mattéo Delabre 2019-12-18 11:01:45 -05:00
parent d8a140d793
commit af8862d360
Signed by: matteo
GPG Key ID: AE3FBD02DC583ABB
1 changed files with 2 additions and 3 deletions

View File

@ -35,9 +35,8 @@ def train_with_parameters(
""" """
model = constellation.ConstellationNet( model = constellation.ConstellationNet(
order=order, order=order,
encoder_layers_sizes=layer_sizes, encoder_layers=layer_sizes,
decoder_layers_sizes=layer_sizes[::-1], decoder_layers=layer_sizes[::-1],
channel_model=constellation.GaussianChannel()
) )
# List of training examples (not shuffled) # List of training examples (not shuffled)