Method __init__

model/downstream.py:9–13

(self, language_model: GLMModel, take_softmax=True, length_penalty=0.0)

class GLMForMultiTokenCloze(torch.nn.Module):
    def __init__(self, language_model: GLMModel, take_softmax=True, length_penalty=0.0):
        super(GLMForMultiTokenCloze, self).__init__()
        self.model = language_model
        self.take_softmax = take_softmax
        self.length_penalty = length_penalty

    def state_dict(self, destination=None, prefix='', keep_vars=False):
        # [h.remove() for h in self.hook_handles]