Skip to content

Commit 1e06f4f

Browse files
Update sub-packages/bionemo-scdl/simple-benchmark/README.md
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: polinabinder1 <pbinder@nvidia.com>
1 parent 97b1a8f commit 1e06f4f

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

  • sub-packages/bionemo-scdl/simple-benchmark

sub-packages/bionemo-scdl/simple-benchmark/README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,12 +106,13 @@ python scdl_speedtest.py --generate-baseline -i my_data.h5ad --scdl-path /path/t
106106
| `--max-time` | Max benchmark runtime (seconds). If the dataset is smaller | 30 |
107107
| `--warmup-time` | Warmup period (seconds). This runs the dataloader before measurement to better reflect average expected performance. | 2 |
108108
| `--csv` | Export detailed CSV files | False |
109+
| `--batch-size` | Batch size used in the PyTorch DataLoader | 64 |
110+
| `--warmup-time` | Warmup period (seconds). This runs the dataloader before measurement to better reflect average expected performance. | 0 |
109111
| `--json` | Export detailed JSON file to specified filename | None |
110112
| `--generate-baseline` | Compare SCDL vs AnnData performance | False |
111113
| `--scdl-path` | Path to SCDL dataset (optional, only used with --generate-baseline) | None |
112114
| `--num-epochs` | The number of epochs (passes through the training dataset). | 1 |
113115
| `--num-runs` | Number of benchmark runs to average (for more stable and reliable measurements) | 1 |
114-
115116
## Sample Output
116117

117118
```

0 commit comments

Comments
 (0)