New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Possible error in reshape? #8

Open

aravindsrinivas opened this issue Oct 28, 2018 · 0 comments

aravindsrinivas commented Oct 28, 2018 •

edited

Loading

self-attention-gan/non_local.py

Line 77 in ad9612e

attn_g = tf.reshape(attn_g, [batch_size, h, w, num_channels // 2])

Is there a mistake in the reshape operation linked above? Shouldn't it be

attn_g = tf.reshape(attn_g, [batch_size, h // 2, w // 2, num_channels // 2])
attn_g = tf.depth_to_space(attn_g, [batch_size, h, w, num_channels // 8])
attn_g = sn_conv1x1(attn_g, num_channels, update_collection, init, 'sn_conv_attn')

instead of attn_g = tf.reshape(attn_g, [batch_size, h, w, num_channels // 2])?

The text was updated successfully, but these errors were encountered:

valillon mentioned this issue

Max pooling and layer dimensions inside attention layer #30

Closed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment