tianyi
/
auditory-visual
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bnDeepNet+SGD

This commit is contained in:
ClF3 2024-10-16 10:44:06 +08:00
parent bbeb915b02
commit 155585271a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
CIFAR10_playground.py
View File

@ -220,8 +220,8 @@ class BnDeepNet(nn.Module):
################### 训练前准备 ################### ################### 训练前准备 ###################
# model = Net('tanh') # model = Net('tanh')
# model = BnNet('relu') # model = BnNet('relu')
model = DeepNet('tanh') # model = DeepNet('tanh')
# model = BnDeepNet('relu') model = BnDeepNet('relu')
device=torch.device("cuda" if torch.cuda.is_available() else "cpu") device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device) model.to(device)