Method train

models/x_mesh_xl/get_model.py:63–68

(self, mode: bool = True)

class MeshXL(nn.Module):
    
    def train(self, mode: bool = True):
        super().train(mode)
        # self.transformer.eval()
        # for param in self.transformer.parameters():
        #     param.requires_grad = False
        return self
    
    def __init__(self, args):
        super().__init__()