(self, name='resnet50', head='linear', feat_dim=128)
| 54 | class RGBMultiHeads(RGBSingleHead): |
| 55 | """RGB model with Multiple linear/mlp projection heads""" |
| 56 | def __init__(self, name='resnet50', head='linear', feat_dim=128): |
| 57 | super(RGBMultiHeads, self).__init__(name, head, feat_dim) |
| 58 | |
| 59 | self.head_jig = JigsawHead(dim_in=int(2048*self.width), |
| 60 | dim_out=feat_dim, |
| 61 | head=head) |
| 62 | |
| 63 | def forward(self, x, x_jig=None, mode=0): |
| 64 | # mode -- |
nothing calls this directly
no test coverage detected