fix: restructuring files
Browse files- model.pt +0 -3
- model.py → model/cnn.py +0 -0
- testbench.ipynb +1 -1
- train.ipynb +1 -1
model.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:58b3f32eee5295b7facfe3c98e6e980c1b27086f483fb7e5414d2e1f648b78e7
|
| 3 |
-
size 4600090
|
|
|
|
|
|
|
|
|
|
|
|
model.py → model/cnn.py
RENAMED
|
File without changes
|
testbench.ipynb
CHANGED
|
@@ -20,7 +20,7 @@
|
|
| 20 |
"import torch\n",
|
| 21 |
"from torch.utils.data import DataLoader\n",
|
| 22 |
"from torchvision import datasets, transforms\n",
|
| 23 |
-
"from model import CNN\n",
|
| 24 |
"\n",
|
| 25 |
"model = CNN()\n",
|
| 26 |
"model.load_state_dict(torch.load(\"model.pt\"))\n",
|
|
|
|
| 20 |
"import torch\n",
|
| 21 |
"from torch.utils.data import DataLoader\n",
|
| 22 |
"from torchvision import datasets, transforms\n",
|
| 23 |
+
"from model.cnn import CNN\n",
|
| 24 |
"\n",
|
| 25 |
"model = CNN()\n",
|
| 26 |
"model.load_state_dict(torch.load(\"model.pt\"))\n",
|
train.ipynb
CHANGED
|
@@ -28,7 +28,7 @@
|
|
| 28 |
"import numpy as np\n",
|
| 29 |
"import matplotlib.pyplot as plt\n",
|
| 30 |
"import torch.nn as nn\n",
|
| 31 |
-
"from model import CNN\n",
|
| 32 |
"from tabulate import tabulate\n",
|
| 33 |
"\n",
|
| 34 |
"\n",
|
|
|
|
| 28 |
"import numpy as np\n",
|
| 29 |
"import matplotlib.pyplot as plt\n",
|
| 30 |
"import torch.nn as nn\n",
|
| 31 |
+
"from model.cnn import CNN\n",
|
| 32 |
"from tabulate import tabulate\n",
|
| 33 |
"\n",
|
| 34 |
"\n",
|