CUDA error: an illegal memory access was encountered while training MatchboxNet (Numba CUDA)
Hello!
I train MatchboxNet model with custom labels and standard config file.
If I use batch_size (train, test, validation) = 512 I consume 3Gb of GPU memory (Tesla V100) and everything works fine. 
If I increase batch_size to 1024 I get ```RuntimeError: CUDA error: an illegal memory access was encountered``` at the first step. 
Running script with ```CUDA_LAUNCH_BLOCKING=1``` gives next output:
```numba.cuda.cudadrv.driver.CudaAPIError: [700] Call to cuLaunchKernel results in UNKNOWN_CUDA_ERROR```
Tried different torch versions - no results.

Any ideas how to fix this?

Numba version: 0.54.0
Python: 3.7.9
CUDA version: 11.1
Driver Version: 455.32.00