Hi Nathan,
I have somethings that “works” mostly. I’m getting a weird CUDA error during training that I’ve asked for some help on here. I haven’t been able to figure it out but it does train on CPU and I am trying to get it work on TPU based on this thread.
The Colab notebook is available here: https://colab.research.google.com/drive/1uK-Wna2Uo9JRx7RqCIrl_W_yVt2tDiBD#offline=true&sandboxMode=true
So far the results are not great, only abut 68%-70% accuracy. I’m hoping that if I can get the training sorted out, I can work on different methods to improve the accuracy.
Hope it can be of some help to you.
Cheers,
Norman