layog/Accurate-Binary-Convolution-Network

How did you implement the STE in equation 10?

wangpichao opened this issue · 3 comments

I read your codes carefully and it is great. And I want to know how did you implement the STE in equation 10?

layog commented

STE is required during the backpropagation only, so it is automatically handled by tensorflow. I implemented the indicator function. You can read how I implemented the indicator function in the text above cell 7 in ABC Notebook (You'll need to download the notebook, as LaTeX is not properly supported by Github)

Thanks your reply@layog. I am not familiar with tensorflow, and if tensorflow could handle the STE, it would be great. I am wondering whether tensorflow could deal with the weights update?

layog commented

Yes, that is what optimizers are for. You can look at the basics of tensorflow on their website.