The sign method would return a zero gradient wouldn’t it?
The only change in the output of sign would be at x=0, so that you might be killing the gradient with this method.
Are you sure the paper uses a sign method for its implementation?
Note that I’m just pointing to “unusual” model design choices, but it might be perfectly valid for your use case.