Merge branch 'main' into dev

micromind-toolkit · Nov 28, 2023 · afbed63 · afbed63
2 parents 9725fdf + d3f9119
commit afbed63
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 12 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -0,0 +1,2 @@
+sphinx
+sphinx_rtd_theme
diff --git a/micromind/__init__.py b/micromind/__init__.py
@@ -2,15 +2,3 @@
 
 # Package version
 __version__ = "0.1.1"
-
-"""datasets_info is a dictionary that contains information about the attributes
-of the datasets.
-This dictionary is used in networks.py inside the from_pretrained class method
-in order to examine the inputs and initialize the PhiNet or, in case of
-mismatching between dataset and Nclasses, raise an AssertionError."""
-datasets_info = {
-    "CIFAR-100": {"Nclasses": 100, "NChannels": 3, "ext": ".pth.tar"},
-    "CIFAR-10": {"Nclasses": 10, "NChannels": 3, "ext": ".pth.tar"},
-    "ImageNet-1k": {"Nclasses": 1000, "NChannels": 3, "ext": ".pth.tar"},
-    "MNIST": {"Nclasses": 10, "NChannels": 1, "ext": ".pth.tar"},
-}