Code link

[Bharath-hn]

There are links only to Trained model. I could not find the code for training or defining the CRU model. Please let me know where can I find the source code.

[cypw]

Thank you for your interest.

You can use mx.model.load_checkpoint to load the complete network definition and use mx.viz.plot_network to visualize it.

As for the code for defining CRU-Net, I was planning to rephrase it and use a better naming rule before making it public but cannot find time to do it. ( check this link for a preliminary version )

[Bharath-hn]

Thank you very much.