fix doc problems

pull/720/head
zhaojichen 5 years ago
parent 4a79dde736
commit e49cd12d92

@ -166,6 +166,8 @@ class _BatchNorm(Cell):
def extend_repr(self):
return 'num_features={}, eps={}, momentum={}, gamma={}, beta={}, moving_mean={}, moving_variance={}'.format(
self.num_features, self.eps, self.momentum, self.gamma, self.beta, self.moving_mean, self.moving_variance)
@constexpr
def _channel_check(channel, num_channel):
if channel != num_channel:
raise ValueError("the input channel is not equal with num_channels")

Loading…
Cancel
Save