def __init__(self, output_dim, body):
    super(VanillaHyperNet, self).__init__()
    self.mixer = False
    self.config = VanillaNet_config(body.feature_dim, output_dim)
    self.fc_head = LinearGenerator(self.config['fc_head'])
    self.body = body
    self.to(Config.DEVICE)
def __init__(self, output_dim, body):
    super(VanillaHyperNet, self).__init__()
    self.mixer = False
    self.config = VanillaNet_config(body.feature_dim, output_dim)
    self.fc_head = LinearGenerator(self.config['fc_head'])
    self.body = body
    self.to(Config.DEVICE)