[BUGFIX] examples/dynamic_pruning.py works again

2021-06-14 20:31:39 +02:00
parent d2856383e2
commit 6197d7d5d6
2 changed files with 2 additions and 2 deletions
--- a/examples/dynamic_pruning.py
+++ b/examples/dynamic_pruning.py
@@ -36,7 +36,7 @@ if __name__ == "__main__":
    # Initialize the model
    model = pt.models.CELVQ(
        hparams,
-        prototype_initializer=pt.components.Ones(2, scale=3),
+        prototypes_initializer=pt.initializers.FVCI(2, 3.0),
    )

    # Compute intermediate input and output sizes