Closed this issue 4 years ago · 1 comments
The norm dim parameter in line 71 of loss.py should be dim=1，in order to calculate norm for per pixels
Awesome thanks. I fixed it and pushed a new version.