changing layers

[mehrnia]

I wanna change some of the layers, what approach you recommend?
I wanna change the unet locally, any advice?

[taigw]

Here is an example of how to use customized layer or network:
https://github.com/HiLab-git/PyMIC_examples/tree/main/seg_full_sup/custom
You can define your own network, and set the network instance for segmentation agent, like this:

    agent  = SegmentationAgent(config, args.stage)
    # use custormized network and loss function
    mynet  = MyUNet2D(config['network'])
    agent.set_network(mynet)
    agent.set_loss_dict(loss_dict)
    agent.run()