marvin/CNN_Website
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor parameter changes

Browse Source
This commit is contained in:
marvin
2026-04-29 20:35:58 +02:00
parent 7f4d2a0cc0
commit acda7e69c0
4 changed files with 36 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bird_cnn/main.py
+3 -3
View File
@@ -34,13 +34,13 @@ val_size = len(dataset) - train_size
train_dataset, val_dataset = random_split(dataset, [train_size, val_size])
train_loader = DataLoader(train_dataset, batch_size=8, shuffle=True)
val_loader = DataLoader(val_dataset, batch_size=8, shuffle=False)
train_loader = DataLoader(train_dataset, batch_size=4, shuffle=True)
val_loader = DataLoader(val_dataset, batch_size=4, shuffle=False)
device = torch.device("cpu") if not torch.cuda.is_available() else torch.device("cuda:0")
print("Using device", device)
model = Bird_CNN(c_in=3, c_hidden=32, c_out=7)
model = Bird_CNN(c_in=3, c_hidden=64, c_out=7)
model.to(device)
optimizer = torch.optim.Adam(model.parameters(), lr=1e-3, weight_decay=1e-4)
loss_module = nn.CrossEntropyLoss()