add data module

m0saan · Jul 3, 2023 · 1a12fec · 1a12fec
1 parent 739c5cc
commit 1a12fec
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/minima/__init__.py b/minima/__init__.py
@@ -8,3 +8,4 @@
 from . import nn
 from . import init
 from . import optim
+from . import data
diff --git a/minima/_modidx.py b/minima/_modidx.py
@@ -95,7 +95,8 @@
                              'minima.data.Dataset.apply_transforms': ('data.html#dataset.apply_transforms', 'minima/data.py'),
                              'minima.data.Sampler': ('data.html#sampler', 'minima/data.py'),
                              'minima.data.Sampler.__init__': ('data.html#sampler.__init__', 'minima/data.py'),
-                             'minima.data.Sampler.__iter__': ('data.html#sampler.__iter__', 'minima/data.py')},
+                             'minima.data.Sampler.__iter__': ('data.html#sampler.__iter__', 'minima/data.py'),
+                             'minima.data.collate': ('data.html#collate', 'minima/data.py')},
             'minima.init': { 'minima.init.constant': ('init.html#constant', 'minima/init.py'),
                              'minima.init.kaiming_normal': ('init.html#kaiming_normal', 'minima/init.py'),
                              'minima.init.kaiming_uniform': ('init.html#kaiming_uniform', 'minima/init.py'),
@@ -271,6 +272,8 @@
                            'minima.nn.Sequential.__iter__': ('nn.html#sequential.__iter__', 'minima/nn.py'),
                            'minima.nn.Sequential.__next__': ('nn.html#sequential.__next__', 'minima/nn.py'),
                            'minima.nn.Sequential.forward': ('nn.html#sequential.forward', 'minima/nn.py'),
+                           'minima.nn.Sigmoid': ('nn.html#sigmoid', 'minima/nn.py'),
+                           'minima.nn.Sigmoid.forward': ('nn.html#sigmoid.forward', 'minima/nn.py'),
                            'minima.nn.Softmax': ('nn.html#softmax', 'minima/nn.py'),
                            'minima.nn.Softmax.forward': ('nn.html#softmax.forward', 'minima/nn.py'),
                            'minima.nn._child_modules': ('nn.html#_child_modules', 'minima/nn.py'),