csrhddlam/axial-deeplab

What's HERE??

wanghao9610 opened this issue 4 years ago · 1 comments

wanghao9610 commented 4 years ago

axial-deeplab/lib/models/utils.py

Line 4 in fe1d052

class qkv_transform(nn.Conv1d):

csrhddlam commented 4 years ago

I believe this is meant for possible customized initializations.

axial-deeplab/lib/models/axialnet.py

Line 176 in fe1d052

if isinstance(m, qkv_transform):

axial-deeplab/lib/models/axialnet.py

Line 90 in fe1d052

self.qkv_transform.weight.data.normal_(0, math.sqrt(1. / self.in_planes))