Rahuletto commited on
Commit
dd07d3c
·
1 Parent(s): 5f81e6e

fix: restructuring files

Browse files
Files changed (4) hide show
  1. model.pt +0 -3
  2. model.py → model/cnn.py +0 -0
  3. testbench.ipynb +1 -1
  4. 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",