paper implement with pytorch

result

32net 32net

16net 16net

8net 8net