[WIP] Update CBC example

examples/cbc_iris.py

@@ -17,18 +17,15 @@ if __name__ == "__main__":
     train_ds = pt.datasets.Iris(dims=[0, 2])
 
     # Reproducibility
-    pl.utilities.seed.seed_everything(seed=3)
+    pl.utilities.seed.seed_everything(seed=42)
 
     # Dataloaders
-    train_loader = torch.utils.data.DataLoader(train_ds,
-                                               num_workers=0,
-                                               batch_size=150)
+    train_loader = torch.utils.data.DataLoader(train_ds, batch_size=32)
 
     # Hyperparameters
     hparams = dict(
-        distribution=[3, 2, 2],
-        proto_lr=0.01,
-        bb_lr=0.01,
+        distribution=[2, 2, 2],
+        proto_lr=0.1,
     )
 
     # Initialize the model
@@ -40,7 +37,7 @@ if __name__ == "__main__":
     # Callbacks
     vis = pt.models.VisCBC2D(data=train_ds,
                              title="CBC Iris Example",
-                             resolution=300,
+                             resolution=100,
                              axis_off=True)
 
     # Setup trainer