Commit 62e24918 authored by Yuxin Wu's avatar Yuxin Wu

fix lint with new flake8 version

parent 20a4204b
...@@ -181,7 +181,7 @@ class SeparateGANTrainer(TowerTrainer): ...@@ -181,7 +181,7 @@ class SeparateGANTrainer(TowerTrainer):
self.tower_func = TowerFuncWrapper(model.build_graph, model.get_inputs_desc()) self.tower_func = TowerFuncWrapper(model.build_graph, model.get_inputs_desc())
with TowerContext('', is_training=True), \ with TowerContext('', is_training=True), \
argscope(BatchNorm, internal_update=True): argscope(BatchNorm, internal_update=True):
# should not hook the updates to both train_op, it will hurt training speed. # should not hook the updates to both train_op, it will hurt training speed.
self.tower_func(*input.get_input_tensors()) self.tower_func(*input.get_input_tensors())
update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS) update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS)
if len(update_ops): if len(update_ops):
......
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