Skip to content

Commit b0df65b

Browse files
committed
Update installation and W&B login instructions in README
1 parent 337ffbe commit b0df65b

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

README.md

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,12 +28,17 @@ A flexible and reusable template for PyTorch-based machine learning experiments.
2828
uv venv
2929
source .venv/bin/activate # On Windows use `.venv\Scripts\activate`
3030
31-
# Install requirements
31+
# Install prerequsites (Recommended)
32+
uv pip install -U torch wandb rich beaupy polars numpy optuna matplotlib scienceplots
33+
34+
# Or sync requirements (caution: this version is optimized for cuda environment)
3235
uv pip sync requirements.txt
36+
3337
# Or using pip: pip install -r requirements.txt
3438
```
3539
4. **(Optional) Login to Weights & Biases:**
3640
```bash
41+
# only once per a machine
3742
wandb login
3843
```
3944
5. **Run a Default Experiment:**
@@ -446,4 +451,3 @@ pruner:
446451
6. When a trial completes, its final validation loss and `pfl` are considered for inclusion in the `top_k` list.
447452
448453
</details>
449-

0 commit comments

Comments
 (0)