@property
def trainable_variables(self):
    return (self.tail.trainable_variables + self.head.trainable_variables)
@property
def trainable_variables(self):
    return (self.tail.trainable_variables + self.head.trainable_variables)