Stabilization Layer

[Marcelo5444]

The stabilization layer were it can be found? is the here:

iColoriT/iColoriT_demo/modeling.py

Line 244 in 8a02415

class CnnHead(nn.Module):

[YeolJ00]

Yes. That is the stabilization layer.

[Marcelo5444]

Hi, checking the stabilizing layer, in our paper diagram its that features forwarded across the Conv2d, keeep the same dimension however, when running the code I see that the:
nn.Conv2d(192, 512, kernel_size=3, stride=1, padding=1, padding_mode='reflect')

Could you explain that, please