Lab3/dataset/ Lab3/mae Lab3/masked Lab3/st Lab3/st2 *.pth .ipynb_checkpoints