Issue about the provided example

[thomascong121]

Hi Thanks for the great work, It is really simple and elegant. But I have a question about the provided example. In your example codes, what is the variable nl_fn in line 70 (last line of function replace_bn)?

[phquang]

Hi, thanks for the interests in our work.
The variable nl_fn should have been nl. I have fixed it in the example.py file.