Update NG

2021-06-07 18:35:08 +02:00
parent d558fa6a4a
commit b031382072
3 changed files with 7 additions and 6 deletions
--- a/prototorch/models/unsupervised.py
+++ b/prototorch/models/unsupervised.py
@@ -28,7 +28,6 @@ class NeuralGas(UnsupervisedPrototypeModel):
        self.save_hyperparameters(hparams)

        # Default hparams
-        self.hparams.setdefault("input_dim", 2)
        self.hparams.setdefault("agelimit", 10)
        self.hparams.setdefault("lm", 1)