Skip to content

Commit d104c8d

Browse files
committed
Expand GUI docs for local and Slurm execution modes
1 parent f072723 commit d104c8d

2 files changed

Lines changed: 25 additions & 2 deletions

File tree

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ Use our pretrained model at [Model Weights](https://huggingface.co/MihinP/PartiN
1717
- Seamless integration with cryoSPARC and RELION workflows
1818
- Confidence-based particle filtering
1919
- Visual detection validation
20+
- Browser GUI (`partinet gui`) with optional Slurm job submission
2021

2122
## Prerequisites
2223

@@ -112,6 +113,7 @@ Available commands:
112113
- `denoise`: Clean input micrographs
113114
- `detect`: Identify particles
114115
- `star`: Generate STAR files
116+
- `gui`: Launch the Gradio GUI (local or Slurm execution)
115117
- `train`: Train custom models
116118

117119

docs/docs/gui.md

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,27 @@ The **Project directory** field at the top of the GUI is set once and auto-fills
5353

5454
Individual fields remain editable if you need to override a path for a specific stage.
5555

56+
Optional environment variables at launch:
57+
58+
- `PARTINET_PROJECT` — pre-fill the project directory
59+
- `PARTINET_WEIGHTS` — pre-fill the Detect weights path
60+
- `PARTINET_SLURM_CONFIG` — path to a user-owned YAML file with default Slurm settings
61+
62+
## Execution modes (Local / Slurm)
63+
64+
Open **Execution settings** to choose how each stage runs:
65+
66+
| Mode | Description |
67+
|------|-------------|
68+
| **Local** | Run `partinet` as a subprocess on the machine hosting the GUI |
69+
| **Slurm** | Write a batch script under `<project>/.partinet_jobs/` and submit with `sbatch` |
70+
71+
All Slurm fields (partition, account, time, CPUs, GPUs, memory) are optional — leave them blank to use your cluster defaults. Use **Job setup script** for site-specific setup (`module load`, `conda activate`, container `apptainer exec`, etc.) without hardcoding paths in the PartiNet source.
72+
73+
Logs stream in the browser from `{project}/partinet_denoise.log`, `partinet_detect.log`, or `partinet_star.log`. Slurm job IDs are shown when a batch job is submitted.
74+
75+
On HPC, run the GUI on a login or visualization node (with SSH port forwarding) and submit Denoise, Detect, and Star File jobs to compute nodes via Slurm.
76+
5677
## Stage 1 · Denoise
5778

5879
Applies the PartiNet Wiener filter denoiser to raw micrographs and saves cleaned images to `project/denoised/`.
@@ -122,8 +143,8 @@ Click **Generate STAR File** to write a CryoSPARC-compatible `.star` file. The c
122143
| Audience | Beginners, interactive exploration | Experienced users, scripting, HPC |
123144
| Log output | Streamed live in the browser | Terminal / log file |
124145
| Threshold tuning | Interactive slider with live preview | Set once at run time |
125-
| Automation | Not suitable | Fully scriptable |
126-
| Container support | Local installation only | Apptainer / Docker |
146+
| Automation | Slurm submission from GUI | Fully scriptable |
147+
| HPC | Login-node GUI + Slurm batch jobs | Apptainer / Docker / modules |
127148

128149
For large-scale or automated processing, the CLI remains the recommended approach.
129150

0 commit comments

Comments
 (0)