Unify stem implementation

Open ziw-liu opened this issue 1 year ago • 1 comments

          @ziw-liu I wrote the `StemDepthtoChannels` to work for resnet and convnext models. Can you test the construction of UNeXt2 model with this stem and report back if the construction works? If it does, we should deprecate other classes for building the stem to avoid confusion in the future.

Originally posted by @mattersoflight in https://github.com/mehta-lab/VisCy/pull/113#discussion_r1699256341

Aug 02 '24 17:08 ziw-liu

https://github.com/mehta-lab/VisCy/pull/113/files#discussion_r1699256669

Aug 02 '24 17:08 ziw-liu

@ziw-liu can viscy.unet.networks.unext2.UNext2Stem be replaced with StemDepthtoChannels in UNext2 architectures?

Aug 27 '24 21:08 mattersoflight

Replacing viscy.unet.networks.unext2.UNext2Stem with StemDepthtoChannels will break the compatibility between the constructed model and checkpoints for the virtual staining task.

Aug 28 '24 00:08 mattersoflight