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).
pip install -r requirements.txt
Run baseline (Fig 10b)
cp params_storage/params_fig10b_single_gpu.py params.py
python train_and_test.py
Run Mamba
cp params_storage/params_mamba_single_gpu.py params.py
python train_and_test.py
Notes
- Outputs go to ./runs//
- This folder is safe to upload to Git/HF (data and runs are ignored).