onnx_program.save("image_classifier_model.onnx")