782 B
782 B
Brittle Star
Usage
UV
To set up the UV module, you can run the following command:
uv sync --frozen
example command:
uv run src/train.py --model_name my_model --epochs 50 --batch_size 32
HPC Integration
To run training on the VSC HPC (Ghent Tier 2):
Setup
Run the installation script once on a login node. This sets up a virtual environment in $VSC_DATA with the required system modules (JAX, Flax, WandB) and lightweight dependencies.
bash scripts/hpc_install.sh
Smoke Test
Verify everything runs on a compute node:
python src/train.py --config configs/hpc_smoke_test.yaml
Submitting Jobs
Submit long-running experiments via the provided Slurm script:
sbatch scripts/hpc_train.slurm