diff --git a/bob/learn/tensorflow/layers/__init__.py b/bob/learn/tensorflow/layers/__init__.py index 7c2ce7f1dfe8fd65131eb4278c62a63d07178183..8cbd8692bb27826b0f8e72d200b431dbd904348d 100644 --- a/bob/learn/tensorflow/layers/__init__.py +++ b/bob/learn/tensorflow/layers/__init__.py @@ -7,6 +7,7 @@ from .Layer import Layer from .Conv2D import Conv2D from .FullyConnected import FullyConnected from .MaxPooling import MaxPooling +from .AveragePooling import AveragePooling from .Dropout import Dropout from .InputLayer import InputLayer diff --git a/bob/learn/tensorflow/network/__init__.py b/bob/learn/tensorflow/network/__init__.py index 0accdd4b1bdff8eee553d8ac1576cbb086bbb918..e583ea984c81c517fa7b201e4bd5459bb147882b 100644 --- a/bob/learn/tensorflow/network/__init__.py +++ b/bob/learn/tensorflow/network/__init__.py @@ -12,6 +12,7 @@ from .MLP import MLP from .FaceNet import FaceNet from .FaceNetSimple import FaceNetSimple from .VGG16 import VGG16 +from .VGG16_mod import VGG16_mod # gets sphinx autodoc done right - don't remove it __all__ = [_ for _ in dir() if not _.startswith('_')]