How to modify the net architecture？

[Wzj02200059]

Hi~ Thanks for this awesome repo。 I have a question，i have noticed that the backbone network used several pooling layer， I think its not conducive to very small size、resolution img， so i just want to reduce the pooling layer and add dilated conv， pls give me some advice that how to modify it。 thanks~

[ChaseMonsterAway]

Hi, could you tell me which config file you want to modify?

[Wzj02200059]

tps_resnet_bilstm_attn.py and small_satrn.py thx

[ChaseMonsterAway]

The params of ConvModule are here. So, if you want to add a dilation params, you can modify a ConvModule as follows:

 ('conv', dict(type='ConvModule', in_channels=1, out_channels=32, kernel_size=3,
               stride=1, padding=1, norm_cfg=norm_cfg, dilation=1))

[Wzj02200059]

OK，i got it