(self, x)
| 141 | # return [scaled_preds, class_preds_lst] if self.training else scaled_preds |
| 142 | |
| 143 | def forward(self, x): |
| 144 | scaled_preds, class_preds = self.forward_ori(x) |
| 145 | class_preds_lst = [class_preds] |
| 146 | return [scaled_preds, class_preds_lst] if self.training else scaled_preds |
| 147 | |
| 148 | |
| 149 | class Decoder(nn.Module): |
nothing calls this directly
no test coverage detected