File size: 971 Bytes
a6b4c35 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | # 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).
|