This makes the short implementation of unet more consistent with the version in cloud_unet. The only difference I kept was that I didn't want to have an n_channels argument, because I think it's more canonical to have the number of channels change at different layers of the net. You can recover the old behavior by setting the filter factors to be constant across the blocks.
This makes the short implementation of unet more consistent with the version in cloud_unet. The only difference I kept was that I didn't want to have an
n_channelsargument, because I think it's more canonical to have the number of channels change at different layers of the net. You can recover the old behavior by setting the filter factors to be constant across the blocks.