paconv_ssg-cuda.py 180 Bytes
Newer Older
1
2
3
4
5
6
7
_base_ = './paconv_ssg.py'

model = dict(
    backbone=dict(
        sa_cfg=dict(
            type='PAConvCUDASAModule',
            scorenet_cfg=dict(mlp_channels=[8, 16, 16]))))