Method __init__

model/downstream.py:115–118

(self, language_model, take_softmax=False)

class GLMForSingleTokenCloze(torch.nn.Module):
    def __init__(self, language_model, take_softmax=False):
        super().__init__()
        self.model = language_model
        self.take_softmax = take_softmax

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