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

changed weight loading

Browse Source
This commit is contained in:
marvin
2026-04-29 14:53:47 +02:00
parent c31abed2f0
commit a9cb14cf42
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bird_cnn/__pycache__/server.cpython-314.pyc
BIN
View File
Binary file not shown.
bird_cnn/server.py
+1 -1
View File
@@ -38,7 +38,7 @@ 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=15, 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, weights_only=False)) model.load_state_dict(torch.load(full_path, map_location=torch.device(device)))
model.to(device) model.to(device)
model.eval() model.eval()