MAD-Clean/pyproject.toml at main · skunkworks-ra/MAD-Clean · GitHub

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
39
40
41
42
43
44
45
46
[project]
name            = "mad-clean"
version         = "0.1.0"
description     = "Morphological Atom Decomposition CLEAN — learned sparse-coding minor cycle for radio interferometric imaging"
readme          = "README.md"
requires-python = ">=3.11"

[tool.pixi.project]
channels  = ["pytorch", "conda-forge"]
platforms = ["linux-64"]

[tool.pixi.dependencies]
python       = ">=3.11"
numpy        = ">=1.26"
scipy        = ">=1.12"
scikit-learn = ">=1.4"
astropy      = ">=6.0"
pytest       = ">=8.0"
matplotlib   = ">=3.8"

[tool.pixi.feature.cpu.dependencies]
pytorch  = {version = ">=2.2", channel = "pytorch"}
cpuonly  = {channel = "pytorch"}

[tool.pixi.feature.gpu.dependencies]
pytorch      = {version = ">=2.2",  channel = "pytorch"}
pytorch-cuda = {version = ">=12.1", channel = "pytorch"}

[tool.pixi.environments]
default = {features = ["cpu"], solve-group = "cpu"}
gpu     = {features = ["gpu"], solve-group = "gpu"}

[tool.pixi.tasks]
fetch         = "python scripts/fetch_crumb.py"
simulate      = "python scripts/simulate_observations.py --data crumb_data/crumb_preprocessed.npz --psf_fwhm 3.0 --noise_std 0.05 --out crumb_data/flow_pairs.npz"
train-patch   = "python scripts/run_train.py --variant A --data crumb_data/flow_pairs.npz --out models/cdl_filters_patch.npy"
train-conv    = "python scripts/run_train.py --variant B --data crumb_data/flow_pairs.npz --out models/cdl_filters_conv.npy --k 32 --atom_size 15"
deconvolve-A      = "python scripts/run_deconvolve.py --variant A --dirty dirty.fits --psf psf.fits --atoms models/cdl_filters_patch.npy --out_dir results/"
deconvolve-B      = "python scripts/run_deconvolve.py --variant B --dirty dirty.fits --psf psf.fits --atoms models/cdl_filters_conv.npy --out_dir results/"
deconvolve-hogbom = "python scripts/run_deconvolve.py --variant hogbom --dirty dirty.fits --psf psf.fits --out_dir results/"
test          = "python -m pytest tests/ -v"

[tool.pixi.feature.gpu.tasks]
train-patch-gpu = "python scripts/run_train.py --variant A --data crumb_data/flow_pairs.npz --out models/cdl_filters_patch.npy --k 128 --lmbda 0.5 --n_epochs 50 --patches_per_img 50 --device cuda"
train-conv-gpu  = "python scripts/run_train.py --variant B --data crumb_data/flow_pairs.npz --out models/cdl_filters_conv.npy --k 128 --lmbda 0.1 --n_epochs 50 --device cuda"
train-flow-gpu  = "python scripts/run_train.py --variant C --data crumb_data/flow_pairs.npz --out models/flow_model.pt --n_epochs 100 --batch_size 8 --lr 1e-4 --device cuda"