better architecture

2026-04-30 18:02:47 +02:00
parent fe7d6221e9
commit da2b9c5052
5 changed files with 2 additions and 2 deletions
@@ -31,7 +31,7 @@ transform = transforms.Compose([

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

-model = Bird_CNN(c_in=3, c_hidden=15, c_out=7)
+model =Bird_CNN(c_in=3, c_hidden=64, c_out=7)
 full_path = os.path.join(BUILD_PATH, "bird_cnn")
 model.load_state_dict(torch.load(full_path, map_location=torch.device(device)))
 model.to(device)