| # DLoc minimal run (baseline + mamba) | |
| This folder is an isolated, minimal run package for: | |
| - DLoc baseline (Fig 10b, single GPU) | |
| - Mamba-DLoc (single GPU) | |
| It excludes all data and training outputs by default. | |
| ## Required data (place in ./data) | |
| - dataset_jacobs_July28.mat (fixed 7,635 samples) | |
| - dataset_non_fov_train_jacobs_July28_2.mat | |
| - dataset_fov_train_jacobs_July28_2.mat | |
| - dataset_non_fov_test_jacobs_July28_2.mat | |
| - dataset_fov_test_jacobs_July28_2.mat | |
| ## Setup | |
| This assumes PyTorch is already installed (e.g., vast.ai PyTorch image). | |
| ```bash | |
| pip install -r requirements.txt | |
| ``` | |
| ## Run baseline (Fig 10b) | |
| ```bash | |
| cp params_storage/params_fig10b_single_gpu.py params.py | |
| python train_and_test.py | |
| ``` | |
| ## Run Mamba | |
| ```bash | |
| cp params_storage/params_mamba_single_gpu.py params.py | |
| python train_and_test.py | |
| ``` | |
| ## Notes | |
| - Outputs go to ./runs/<timestamp>/ | |
| - This folder is safe to upload to Git/HF (data and runs are ignored). | |