Method train

models/x_mesh_xl/get_model.py:12–17

(self, mode: bool = True)

class ConditionEncoder(nn.Module):

    def train(self, mode: bool = True):
        super().train(mode)
        self.multi_encoder.eval()
        for param in self.multi_encoder.parameters():
            param.requires_grad = False
        return self

    def __init__(self, args, hidden_size):
        super().__init__()