MeshDiffusion/data/grid_mask_64.pt