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

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)