Welcome to the forum @Yehuda .
This job may be running out of GPU memory. Memory use depends, among other parameters, on
- the number of classes
- the box size
Please can you share those parameters, along with information listed in the Environment section of the Troubleshooting Guidelines.