remove blocks

ASEM000 · Aug 28, 2023 · 584cf53 · 584cf53
1 parent 5e6ae12
commit 584cf53
Show file tree

Hide file tree

Showing 10 changed files with 5 additions and 479 deletions.
diff --git a/CHANEGLOG.md b/CHANEGLOG.md
@@ -55,3 +55,4 @@
 
 - `Bilinear` is deprecated, use `Multilinear((in1_features, in2_features), out_features)`
 - `HistogramEqualization2D`
+- Remove `.blocks`, and will move it to examples
diff --git a/docs/API/api.rst b/docs/API/api.rst
@@ -24,6 +24,5 @@
    pooling
    recurrent
    reshaping
-   misc
 
 
diff --git a/docs/API/containers.rst b/docs/API/containers.rst
@@ -4,4 +4,5 @@ Containers
 
 
 .. autoclass:: Sequential
-.. autoclass:: RandomApply
+.. autoclass:: RandomApply
+.. autoclass:: RandomChoice
diff --git a/docs/API/misc.rst b/docs/API/misc.rst
diff --git a/serket/nn/__init__.py b/serket/nn/__init__.py
@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from . import blocks
 from .activation import (
     ELU,
     GELU,
@@ -46,7 +45,6 @@
     ThresholdedReLU,
 )
 from .attention import MultiHeadAttention
-from .blocks import UNetBlock, VGG16Block, VGG19Block
 from .clustering import KMeans
 from .containers import RandomApply, RandomChoice, Sequential
 from .convolution import (
@@ -214,10 +212,6 @@
     "ThresholdedReLU",
     # attention
     "MultiHeadAttention",
-    # blocks
-    "UNetBlock",
-    "VGG16Block",
-    "VGG19Block",
     # container
     "RandomApply",
     "RandomChoice",

diff --git a/serket/nn/blocks/__init__.py b/serket/nn/blocks/__init__.py
diff --git a/serket/nn/blocks/unet.py b/serket/nn/blocks/unet.py