Skip to content

Instantly share code, notes, and snippets.

Last active August 19, 2020 08:51
What would you like to do?
# Bad practice, don't return tuple
class RetinaNet(nn.Module):
def forward(self, image):
x = self.encoder(image)
x = self.decoder(x)
bboxes, scores = self.head(x)
return bboxes, scores
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment