class_docstr: ResNet head.
class_docstr_tok: ["ResNet","head","."]
code: def __init__(self, w_in, net_params):
      super(ResHead, self).__init__()
      self.avg_pool = nn.AdaptiveAvgPool2d((1, 1))
      self.use_dropout = net_params["use_dropout"]
      if self.use_dropout:
          self.dropout = nn.Dropout(
              p=net_params["dropout_rate"], inplace=True
          )
      self.fc = nn.Linear(w_in, net_params["nc"], bias=True)
  
imports: import numpy
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 to create a new ResHead object from inherited base classes: nn with example: obj = ResHead(w_in, net_params)