Update example scripts

examples/dynamic_pruning.py

@@ -30,7 +30,7 @@ if __name__ == "__main__":
     prototypes_per_class = num_clusters * 5
     hparams = dict(
         distribution=(num_classes, prototypes_per_class),
-        lr=0.1,
+        lr=0.2,
     )
 
     # Initialize the model
@@ -39,6 +39,12 @@ if __name__ == "__main__":
         prototype_initializer=pt.components.Ones(2, scale=3),
     )
 
+    # Compute intermediate input and output sizes
+    model.example_input_array = torch.zeros(4, 2)
+
+    # Summary
+    print(model)
+
     # Callbacks
     vis = pt.models.VisGLVQ2D(train_ds)
     pruning = pt.models.PruneLoserPrototypes(
@@ -67,7 +73,7 @@ if __name__ == "__main__":
         ],
         progress_bar_refresh_rate=0,
         terminate_on_nan=True,
-        weights_summary=None,
+        weights_summary="full",
         accelerator="ddp",
     )