Update README.md
Browse files
README.md
CHANGED
|
@@ -127,6 +127,7 @@ print(dataset['train'][322]['transcription'].lower())
|
|
| 127 |
```python
|
| 128 |
import gc
|
| 129 |
import torch
|
|
|
|
| 130 |
|
| 131 |
# Free up unused memory in CUDA (GPU) - only needed if you use a GPU
|
| 132 |
if torch.cuda.is_available():
|
|
|
|
| 127 |
```python
|
| 128 |
import gc
|
| 129 |
import torch
|
| 130 |
+
import psutil
|
| 131 |
|
| 132 |
# Free up unused memory in CUDA (GPU) - only needed if you use a GPU
|
| 133 |
if torch.cuda.is_available():
|