fix reference

ziw-liu · ziw-liu · commit 2eb7c221f5ce · 2024-05-25T12:33:00.000+08:00
diff --git a/viscy/unet/networks/Unet22D.py b/viscy/unet/networks/Unet22D.py
@@ -316,7 +316,7 @@ def __init__(
         # replace first convolution layer with a projection tokenizer
         multi_scale_encoder.stem_0 = nn.Identity()
         self.encoder_stages = multi_scale_encoder
-        self.stem = Conv21dStem(
+        self.stem = Conv22dStem(
             in_channels, num_channels[0], stem_kernel_size, in_stack_depth
         )
         decoder_channels = num_channels

Original file line number	Diff line number	Diff line change
`@@ -316,7 +316,7 @@ def __init__(`
`316`	`316`	`# replace first convolution layer with a projection tokenizer`
`317`	`317`	`multi_scale_encoder.stem_0 = nn.Identity()`
`318`	`318`	`self.encoder_stages = multi_scale_encoder`
`319`		`- self.stem = Conv21dStem(`
	`319`	`+ self.stem = Conv22dStem(`
`320`	`320`	`in_channels, num_channels[0], stem_kernel_size, in_stack_depth`
`321`	`321`	`)`
`322`	`322`	`decoder_channels = num_channels`