Skip to content

Commit

Permalink
fix comment
Browse files Browse the repository at this point in the history
  • Loading branch information
yfeng95 committed Apr 27, 2018
1 parent d16fbaa commit c358444
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import numpy as np

def resBlock(x, num_outputs, kernel_size = 4, stride=1, activation_fn=tf.nn.relu, normalizer_fn=tcl.batch_norm, scope=None):
assert num_outputs%2==0, "num_outputs must be divided by channel_factor %d." % self.channel_factor
assert num_outputs%2==0 #num_outputs must be divided by channel_factor(2 here)
with tf.variable_scope(scope, 'resBlock'):
shortcut = x
if stride != 1 or x.get_shape()[3] != num_outputs:
Expand All @@ -16,7 +16,6 @@ def resBlock(x, num_outputs, kernel_size = 4, stride=1, activation_fn=tf.nn.relu

x += shortcut
x = normalizer_fn(x)

x = activation_fn(x)
return x

Expand Down

0 comments on commit c358444

Please sign in to comment.