edited some comments

Author: PineapplePulp

examples/cat_breeds/models/for_cats.py

@@ -1,6 +1,4 @@
-import torch
 import torch.nn as nn
-import torch.nn.functional as F
 
 class SmallCNN(nn.Module):
     def __init__(self):
@@ -17,8 +15,4 @@ def __init__(self):
         )
 
     def forward(self, x):
-        return self.layers(x)
-    
-    # NOTES
-    # Must use nn.Flatten instead of x.view()
-    # Must use pool1 and pool2 rather than one pool.
+        return self.layers(x)

examples/cat_breeds/to_chai.py

@@ -10,11 +10,11 @@
 model = torch.load("./cat_breeds/models/pretest.pt")
 model.chai_dump("./cat_breeds/models/chai_model", "SmallCNN")
 
-# load_path = "./cat_breeds/data/catbreeds/images"
-# for i, item in enumerate(os.listdir(load_path)):
-#     if "item" in item: # check file name
-#         img = np.load(f"{load_path}/{item}")
-#         img = torch.Tensor(img)
-#         img.chai_save("./cat_breeds/data/catbreeds/chai_images", f"item{i}", verbose=False)
-#     if i > 20:
-#         break
+load_path = "./cat_breeds/data/catbreeds/images"
+for i, item in enumerate(os.listdir(load_path)):
+    if "item" in item: # check file name
+        img = np.load(f"{load_path}/{item}")
+        img = torch.Tensor(img)
+        img.chai_save("./cat_breeds/data/catbreeds/chai_images", f"item{i}", verbose=False)
+    if i > 20:
+        break