Skip to content

No weight update due to sign function #6

@jonkoi

Description

@jonkoi

Hi,

I don't think weights get updated because of the tf.sign function cutting off backprop gradients:

binarized_filters = tf.sign(tiled_filters + expanded_stddev, name="binarized_filters")

i believe this line is needed:

with tf.get_default_graph().gradient_override_map({"Sign": "Identity"}):

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions