examples/research_projects/pplm/run_pplm_discrim_train.py/Discriminator/train_custom
class Discriminator:    def train_custom(self):
        for param in self.encoder.parameters():
            param.requires_grad = False
        self.classifier_head.train()