original_docstring: 
          No sigmoid in forward because we are going to
          use BCEWithLogitsLoss
          Which applies sigmoid for us when calculating
          a loss
          
docstring: No sigmoid in forward because we are going to
  use BCEWithLogitsLoss
  Which applies sigmoid for us when calculating
  a loss
docstring_tokens: No sigmoid in forward because we are going to use BCEWithLogitsLoss Which applies sigmoid for us when calculating a loss
code: def forward(self, images):
          """
          No sigmoid in forward because we are going to
          use BCEWithLogitsLoss
          Which applies sigmoid for us when calculating
          a loss
          """
          x = images
          features = self.arch(x)
          output = self.ouput(features)
          if self.return_feats:
              return features
          return output
code_tokens: def forward ( self , images ) : x = images features = self . arch ( x ) output = self . ouput ( features ) if self . return_feats : return features return output
short_docstring: No sigmoid in forward because we are going to
  use BCEWithLogitsLoss
  Which applies sigmoid for us when calculating
  a loss
short_docstring_tokens: No sigmoid in forward because we are going to use BCEWithLogitsLoss Which applies sigmoid for us when calculating a loss
comment: """
          No sigmoid in forward because we are going to
          use BCEWithLogitsLoss
          Which applies sigmoid for us when calculating
          a loss
          """