File tree Expand file tree Collapse file tree
sub-packages/bionemo-scdl/simple-benchmark Expand file tree Collapse file tree Original file line number Diff line number Diff 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```
You can’t perform that action at this time.
0 commit comments