From e54bf07030846216948c8b9f7d2d0d4e76d6c983 Mon Sep 17 00:00:00 2001 From: blackfly Date: Sat, 11 Apr 2020 17:35:00 +0200 Subject: [PATCH] Populate init files --- prototorch/functions/__init__.py | 12 ++++++++++++ prototorch/modules/__init__.py | 7 +++++++ 2 files changed, 19 insertions(+) diff --git a/prototorch/functions/__init__.py b/prototorch/functions/__init__.py index e69de29..d69d07c 100644 --- a/prototorch/functions/__init__.py +++ b/prototorch/functions/__init__.py @@ -0,0 +1,12 @@ +"""ProtoTorch functions.""" + +from .activations import identity, sigmoid_beta, swish_beta +from .competitions import knnc, wtac + +__all__ = [ + 'identity', + 'sigmoid_beta', + 'swish_beta', + 'knnc', + 'wtac', +] diff --git a/prototorch/modules/__init__.py b/prototorch/modules/__init__.py index e69de29..fcaffe1 100644 --- a/prototorch/modules/__init__.py +++ b/prototorch/modules/__init__.py @@ -0,0 +1,7 @@ +"""ProtoTorch modules.""" + +from .prototypes import Prototypes1D + +__all__ = [ + 'Prototypes1D', +]