Example #1
0
 def get_res_block(in_channels, out_channels, padding):
     return blocks.ResBlock(in_channels,
                            out_channels,
                            padding,
                            upsample=False,
                            downsample=False,
                            norm_layer='none')
Example #2
0
 def get_up_block(in_channels, out_channels, padding, norm_layer):
     return blocks.ResBlock(in_channels,
                            out_channels,
                            padding,
                            upsample=True,
                            downsample=False,
                            norm_layer=norm_layer)