I'm currently working on a MAXIM integrated AI project. For more information please check this Github link: https://github.com/MaximIntegratedAI/ai8x-synthesis I've been following the steps mentioned in the README file and reached the Model Training and Quantization section.To train the FP32 model for MNIST on MAX78000, I ran the following command: bash scripts/train_mnist.sh
But I keep getting the error: RuntimeError: CUDA error: no kernel image is available for execution on the device
Any guidance or hints towards fixing the problem will be of great help. Thanks!