Hi @BobMcDear great answer. I asked the FastAI discord server and found very interesting resources (thanks to discord user @ Roshi).
Basically they guide you on how to convert the pytorch model with ONNX so you can use it with Coral TPU.
As far as I have read (but I am a very beginner), ONNX is not very complex to use to convert pytorch models but I am probably wrong for what you say.
Example code I found to process Pytorch model:
// Code written by @ Roshi
torch.onnx.export(
learn.model, # the model
torch.randn(1, 3, 224, 224, device='cpu'), #Dummy input
"PATH_TO_PRINT_MODEL", #Name your model and where you want to put it
input_names=["input"], # No clue
output_names=['output'] #Yes...
)
I think NVIDIA products may be more suitable in this case. I will have to study jetson nano in more detail. Do you know if its compatible with Raspberry Pi or do I need the developer kit?
And thank you very much.