Method
__init__
(
self,
channel_shared=False,
in_channels=None,
a_init=truncated_normal(mean=0.0, stddev=0.05),
name=None # "ptrelu6"
)
Source from the content-addressed store, hash-verified
| 229 | """ |
| 230 | |
| 231 | def __init__( |
| 232 | self, |
| 233 | channel_shared=False, |
| 234 | in_channels=None, |
| 235 | a_init=truncated_normal(mean=0.0, stddev=0.05), |
| 236 | name=None # "ptrelu6" |
| 237 | ): |
| 238 | |
| 239 | super(PTRelu6, self).__init__(name) |
| 240 | self.channel_shared = channel_shared |
| 241 | self.in_channels = in_channels |
| 242 | self.a_init = a_init |
| 243 | |
| 244 | if self.channel_shared: |
| 245 | self.build((None, )) |
| 246 | self._built = True |
| 247 | elif self.in_channels: |
| 248 | self.build((None, self.in_channels)) |
| 249 | self._built = True |
| 250 | |
| 251 | logging.info("PTRelu6 %s: channel_shared: %s" % (self.name, self.channel_shared)) |
| 252 | |
| 253 | def __repr__(self): |
| 254 | s = ('{classname}(') |
Callers
nothing calls this directly
Tested by
no test coverage detected