Method freeze

sat/sgm/modules/encoders/modules.py:209–213

(self)

            self.freeze()

    def freeze(self):
        self.transformer = self.transformer.eval()

        for param in self.parameters():
            param.requires_grad = False

    # @autocast
    def forward(self, text):