[Lesson6 - CustomUnet]
Hi! I have a question according your custom UNET. After adding the new head to the model, you initialize the layers with apply_init(nn.Sequential(layers[3], layers[-2]), init)
Why do you pass in this new Sequential-model: nn.Sequential(layers[3], layers[-2])? I am a bit confused with the indices. Which layers do you address?