Update model.py

oriolcolomefont · Jan 28, 2024 · a963cac · a963cac
1 parent c9e1949
commit a963cac
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/model.py b/model.py
@@ -1,3 +1,7 @@
+"""
+The SampleCNN class is as implemented by https://github.com/Spijkervet/CLMR 
+"""
+
 # Import the required libraries
 import torch
 import torch.nn as nn