GlobalAveragePooling: convert to tf.layers data_format (#678)

26d792d4 · Richard Muñoz · Yuxin Wu · a759dbb0 · 26d792d4
Commit 26d792d4 authored Feb 27, 2018 by Richard Muñoz Committed by Yuxin Wu Feb 27, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tensorpack/models/pool.py tensorpack/models/pool.py +1 -0

No files found.
--- a/tensorpack/models/pool.py
+++ b/tensorpack/models/pool.py
@@ -57,6 +57,7 @@ def AvgPooling(


 @layer_register(log_shape=True)
+@convert_to_tflayer_args(args_names=[], name_mapping={})
 def GlobalAvgPooling(x, data_format='channels_last'):
    """
    Global average pooling as in the paper `Network In Network