(self, output, mask, ind, rotbin, rotres)
| 196 | super(BinRotLoss, self).__init__() |
| 197 | |
| 198 | def forward(self, output, mask, ind, rotbin, rotres): |
| 199 | pred = gather_feature(output, ind, use_transform=True) |
| 200 | loss = compute_rot_loss(pred, rotbin, rotres, mask) |
| 201 | return loss |
| 202 | |
| 203 | |
| 204 | def compute_res_loss(output, target): |
nothing calls this directly
no test coverage detected