Skip to content

Commit f5f0362

Browse files
committed
add note of what OOM error looks like during training
1 parent c0156c9 commit f5f0362

File tree

1 file changed

+1
-1
lines changed
  • AI-and-Analytics/End-to-end-Workloads/LanguageIdentification

1 file changed

+1
-1
lines changed

AI-and-Analytics/End-to-end-Workloads/LanguageIdentification/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ The YAML file `train_ecapa.yaml` with the training configurations is passed as a
203203
| `seed` | The seed value, which should be set to a different value for subsequent runs. Defaults to 1987.
204204
| `out_n_neurons` | Must be equal to the number of languages of interest.
205205
| `number_of_epochs` | Default is **10**. Adjust as needed.
206-
| `batch_size` | In the trainloader_options, decrease this value if your CPU or GPU runs out of memory while running the training script.
206+
| `batch_size` | In the trainloader_options, decrease this value if your CPU or GPU runs out of memory while running the training script. If you see a "Killed" error message, then the training script has run out of memory.
207207

208208
2. Run the script to train the model.
209209
```

0 commit comments

Comments
 (0)