Browse Source

Reflect parameter changes in `experiment.py`

master
Mattéo Delabre 3 years ago
parent
commit
af8862d360
Signed by: matteo GPG Key ID: AE3FBD02DC583ABB
  1. 5
      experiment.py

5
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)

Loading…
Cancel
Save