test(githooks): Add githooks for automatic commit checks

examples/cli/README.md

@@ -5,4 +5,4 @@ Examples in this folder use the experimental [Lightning CLI](https://pytorch-lig
 To use the example run
 ```
 python gmlvq.py --config gmlvq.yaml
-```
+```

examples/cli/gmlvq.py

@@ -1,12 +1,20 @@
 """GMLVQ example using the MNIST dataset."""
 
-from prototorch.models import ImageGMLVQ
-from prototorch.models.data import train_on_mnist
+import torch
 from pytorch_lightning.utilities.cli import LightningCLI
 
-
-class GMLVQMNIST(train_on_mnist(batch_size=64), ImageGMLVQ):
-    """Model Definition."""
+import prototorch as pt
+from prototorch.models import ImageGMLVQ
+from prototorch.models.abstract import PrototypeModel
+from prototorch.models.data import MNISTDataModule
 
 
-cli = LightningCLI(GMLVQMNIST)
+class ExperimentClass(ImageGMLVQ):
+    def __init__(self, hparams, **kwargs):
+        super().__init__(hparams,
+                         optimizer=torch.optim.Adam,
+                         prototype_initializer=pt.components.zeros(28 * 28),
+                         **kwargs)
+
+
+cli = LightningCLI(ImageGMLVQ, MNISTDataModule)

examples/cli/gmlvq.yaml

@@ -1,9 +1,11 @@
 model:
   hparams:
+    input_dim: 784
+    latent_dim: 784
     distribution:
       num_classes: 10
       prototypes_per_class: 2
-    input_dim: 784
-    latent_dim: 784
     proto_lr: 0.01
     bb_lr: 0.01
+data:
+  batch_size: 32

examples/dynamic_pruning.py

@@ -2,10 +2,11 @@
 
 import argparse
 
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 
+import prototorch as pt
+
 if __name__ == "__main__":
     # Command-line arguments
     parser = argparse.ArgumentParser()

examples/glvq_iris.py

@@ -2,11 +2,12 @@
 
 import argparse
 
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 from torch.optim.lr_scheduler import ExponentialLR
 
+import prototorch as pt
+
 if __name__ == "__main__":
     # Command-line arguments
     parser = argparse.ArgumentParser()

examples/glvq_spiral.py

@@ -2,10 +2,11 @@
 
 import argparse
 
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 
+import prototorch as pt
+
 if __name__ == "__main__":
     # Command-line arguments
     parser = argparse.ArgumentParser()

examples/gng_iris.py

@@ -2,10 +2,11 @@
 
 import argparse
 
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 
+import prototorch as pt
+
 if __name__ == "__main__":
     # Command-line arguments
     parser = argparse.ArgumentParser()

examples/knn_iris.py

@@ -2,11 +2,12 @@
 
 import argparse
 
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 from sklearn.datasets import load_iris
 
+import prototorch as pt
+
 if __name__ == "__main__":
     # Command-line arguments
     parser = argparse.ArgumentParser()

examples/ksom_colors.py

@@ -2,11 +2,12 @@
 
 import argparse
 
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 from matplotlib import pyplot as plt
 
+import prototorch as pt
+
 
 def hex_to_rgb(hex_values):
     for v in hex_values:

examples/lgmlvq_moons.py

@@ -2,10 +2,11 @@
 
 import argparse
 
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 
+import prototorch as pt
+
 if __name__ == "__main__":
     # Command-line arguments
     parser = argparse.ArgumentParser()

examples/liramlvq_tecator.py

@@ -3,10 +3,11 @@
 import argparse
 
 import matplotlib.pyplot as plt
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 
+import prototorch as pt
+
 
 def plot_matrix(matrix):
     title = "Lambda matrix"

examples/ng_iris.py

@@ -2,13 +2,14 @@
 
 import argparse
 
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 from sklearn.datasets import load_iris
 from sklearn.preprocessing import StandardScaler
 from torch.optim.lr_scheduler import ExponentialLR
 
+import prototorch as pt
+
 if __name__ == "__main__":
     # Command-line arguments
     parser = argparse.ArgumentParser()

examples/rslvq_iris.py

@@ -2,11 +2,12 @@
 
 import argparse
 
-import prototorch as pt
 import pytorch_lightning as pl
 import torch
 from torchvision.transforms import Lambda
 
+import prototorch as pt
+
 if __name__ == "__main__":
     # Command-line arguments
     parser = argparse.ArgumentParser()