(self, x, head_only=False, loc=True)
| 144 | self.p_head = p_head |
| 145 | |
| 146 | def forward(self, x, head_only=False, loc=True): |
| 147 | output = self.backbone(x) |
| 148 | logits = self.ca_head(output, loc) |
| 149 | return [self.c_head(logits[:,:1]), self.p_head(logits[:,1:])] |
| 150 | |
| 151 | def train_dino(args): |
| 152 | # os.environ['MASTER_PORT'] = '30000' |
nothing calls this directly
no outgoing calls
no test coverage detected