From 155585271aec4a005c4dc427fc2225aecd34ae97 Mon Sep 17 00:00:00 2001
From: ClF3 <smh66ccff@outlook.com>
Date: Wed, 16 Oct 2024 10:44:06 +0800
Subject: [PATCH] bnDeepNet+SGD

---
 CIFAR10_playground.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/CIFAR10_playground.py b/CIFAR10_playground.py
index 404637e..4f01afa 100755
--- a/CIFAR10_playground.py
+++ b/CIFAR10_playground.py
@@ -220,8 +220,8 @@ class BnDeepNet(nn.Module):
 ################### 训练前准备 ###################
 # model = Net('tanh')
 # model = BnNet('relu')
-model = DeepNet('tanh')
-# model = BnDeepNet('relu')
+# model = DeepNet('tanh')
+model = BnDeepNet('relu')
 
 device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)