I have 3 questions about the diagram:
In the diagram, all layers except the last one are represented as
ResidualConvBlocks, but in the code, only the first layer (
Init.conv) actually sets
is_resto true. Should all other except
Init.convhave a different icon/shape to reflect that they are in fact not residual?
In the diagram,
up0is represented in the same way as
UnetUp, with 1
ResidualConvBlocks. However in code it is only a single
ConvTranspose2dwith a group norm. Should it have a different icon/shape in the diagram?
The last layer in the diagram is represented as two
ConvTranspose2d, but in code they are
Conv2d. Should their color be bright blue instead of dark blue?