EasyOCR Training Issue
nrosto opened this issue · 1 comments
nrosto commented
When using images with a height of 64 pixels and setting imageH = 64, everything crashes.
However, when setting the image height to 32 pixels and using images with a height of 32 pixels, the training process proceeds successfully.
I also tried training with image height set to 32 pixels in config.yaml, but it crashes.
Has anyone encountered this?
nrosto commented
Fixed with:
changed batch_size from 64 to 16
but i don't know how it's work
I get
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.104.12 Driver Version: 535.104.12 CUDA Version: 12.2 |
|-----------------------------------------+----------------------+----------------------+
| GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|=========================================+======================+======================|
| 0 NVIDIA RTX A5000 On | 00000000:01:00.0 Off | Off |
| 30% 27C P8 18W / 230W | 3732MiB / 24564MiB | 0% Default |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+
| 1 NVIDIA RTX A5000 On | 00000000:24:00.0 Off | Off |
| 30% 27C P8 18W / 230W | 5MiB / 24564MiB | 0% Default |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+
| 2 NVIDIA RTX A5000 On | 00000000:41:00.0 Off | Off |
| 30% 27C P8 20W / 230W | 5MiB / 24564MiB | 0% Default |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+
| 3 NVIDIA RTX A5000 On | 00000000:61:00.0 Off | Off |
| 30% 26C P8 17W / 230W | 5MiB / 24564MiB | 0% Default |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+
| 4 NVIDIA RTX A5000 On | 00000000:A1:00.0 Off | Off |
| 30% 25C P8 19W / 230W | 5MiB / 24564MiB | 0% Default |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+