<img width="1072" alt="image" src="https://github.com/mlpc-ucsd/Patch-DM/assets/44183898/344382df-839a-4537-ab11-3b5cfa90d9fa"> I try to run train_latent on my custom dataset but I run into NameError: free variable 'patch_num_x' referenced before assignment in enclosing scope