We should create a script that tests the model by just running the compiled onnx or the last checkpoint on a map