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

better architecture

Browse Source
This commit is contained in:
marvin
2026-04-30 18:02:47 +02:00
parent fe7d6221e9
commit da2b9c5052
5 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bird_cnn/__pycache__/bird_cnn.cpython-314.pyc
BIN
View File
Binary file not shown.
bird_cnn/__pycache__/server.cpython-314.pyc
BIN
View File
Binary file not shown.
bird_cnn/build_models/bird_cnn
BIN
View File
Binary file not shown.
bird_cnn/server.py
+1 -1
View File
@@ -31,7 +31,7 @@ transform = transforms.Compose([
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 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") full_path = os.path.join(BUILD_PATH, "bird_cnn")
model.load_state_dict(torch.load(full_path, map_location=torch.device(device))) model.load_state_dict(torch.load(full_path, map_location=torch.device(device)))
model.to(device) model.to(device)