自定义Pytorch Cifar_10 Dataset

MansourGu · Oct 21, 2022 · 8e99096 · 8e99096
1 parent 011b2e5
commit 8e99096
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/AlexNet/AlexNet_Model_Pytorch.py b/AlexNet/AlexNet_Model_Pytorch.py
@@ -1,2 +1,4 @@
+from torch import nn
+