Skip to content

S
seminar-breakout
You need to sign in or sign up before continuing.
Commit 9f056711 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

strict LinearWrap

parent c24697ee
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 7 deletions
tensorpack/models/linearwrap.py
View file @ 9f056711
...@@ -5,7 +5,6 @@ ...@@ -5,7 +5,6 @@
import six import six
from types import ModuleType from types import ModuleType
from ..utils import logger
from .common import get_registered_layer from .common import get_registered_layer
__all__ = ['LinearWrap'] __all__ = ['LinearWrap']
...@@ -58,12 +57,11 @@ class LinearWrap(object): ...@@ -58,12 +57,11 @@ class LinearWrap(object):
return LinearWrap(ret) return LinearWrap(ret)
return f return f
else: else:
if layer_name != 'tf': assert layer_name == 'tf', \
logger.warn( "Calling LinearWrap.{}:" \
"You're calling LinearWrap.__getattr__ with {}:" " neither a layer nor 'tf'! " \
" neither a layer nor 'tf'!".format(layer_name)) "Did you forget to extract tensor from LinearWrap?".format(layer_name)
import tensorflow as tf # noqa import tensorflow as layer # noqa
layer = eval(layer_name)
assert isinstance(layer, ModuleType), layer assert isinstance(layer, ModuleType), layer
return LinearWrap._TFModuleFunc(layer, self._t) return LinearWrap._TFModuleFunc(layer, self._t)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment