Method state_dict

model/model_utils.py:554–560

(self, **kwargs)

        self.encoder_proj = nn.Linear(192, 192, bias=False)  # hard code for now
    
    def state_dict(self, **kwargs):
        # remove the condition encoder from the state dict
        state_dict = super().state_dict(**kwargs)
        for k in list(state_dict.keys()):
            if "condition_encoder" in k:
                del state_dict[k]
        return state_dict
    
    def train(self, mode=True):
        super().train(mode)