onnx

1
{
2
 "cells": [
3
  {
4
   "cell_type": "code",
5
   "execution_count": 1,
6
   "metadata": {},
7
   "outputs": [
8
    {
9
     "name": "stdout",
10
     "output_type": "stream",
11
     "text": [
12
      "The model is saved.\n"
13
     ]
14
    }
15
   ],
16
   "source": [
17
    "import onnx\n",
18
    "import os\n",
19
    "\n",
20
    "\n",
21
    "# Preprocessing: load the old model\n",
22
    "old_model_path = os.path.join(\"resources\", \"single_relu.onnx\")\n",
23
    "onnx_model = onnx.load(old_model_path)\n",
24
    "\n",
25
    "# Preprocessing: get the path to the saved model\n",
26
    "new_model_path = os.path.join(\"resources\", \"single_relu_new.onnx\")\n",
27
    "\n",
28
    "# Save the ONNX model\n",
29
    "onnx.save(onnx_model, new_model_path)\n",
30
    "\n",
31
    "print(\"The model is saved.\")"
32
   ]
33
  }
34
 ],
35
 "metadata": {
36
  "kernelspec": {
37
   "display_name": "Python 3",
38
   "language": "python",
39
   "name": "python3"
40
  },
41
  "language_info": {
42
   "codemirror_mode": {
43
    "name": "ipython",
44
    "version": 3
45
   },
46
   "file_extension": ".py",
47
   "mimetype": "text/x-python",
48
   "name": "python",
49
   "nbconvert_exporter": "python",
50
   "pygments_lexer": "ipython3",
51
   "version": "3.9.2"
52
  }
53
 },
54
 "nbformat": 4,
55
 "nbformat_minor": 2
56
}
57