diff --git a/module/resnet_block.py b/module/resnet_block.py index e65258e..e4eeeb6 100644 --- a/module/resnet_block.py +++ b/module/resnet_block.py @@ -91,8 +91,10 @@ def build_conv_block(self, dim, padding_type, norm_layer, use_dropout, use_bias) # First conv layer padding if padding_type == 'reflect': + # Use reflection padding, which mirrors the border pixels for padding conv_block += [nn.ReflectionPad2d(1)] elif padding_type == 'replicate': + # Use replication padding, which replicates the border pixels for padding conv_block += [nn.ReplicationPad2d(1)] elif padding_type == 'zero': p = 1 # Use padding in conv layer instead