Re-implementation in PyTorch

[matteo-rizzo]

Hi! I'm working on an open-source re-implementation of the FC4 method in PyTorch. The code is available at https://github.com/matteo-rizzo/pytorch-fc4. Everything seems to work quite smoothly, but I would much appreciate it if @yuanming-hu or anyone who is interested in the project could have a look.

• The architecture is defined at https://github.com/matteo-rizzo/pytorch-fc4/blob/main/classes/fc4/FC4.py
• Running the visualize.py script allows to generate this kind of plots:

Thank you!

[flyingwaterdog]

hello，why correction image looks so red, is it normal?