diff --git a/bird_cnn/__pycache__/bird_cnn.cpython-314.pyc b/bird_cnn/__pycache__/bird_cnn.cpython-314.pyc index 927c59c..b941511 100644 Binary files a/bird_cnn/__pycache__/bird_cnn.cpython-314.pyc and b/bird_cnn/__pycache__/bird_cnn.cpython-314.pyc differ diff --git a/bird_cnn/__pycache__/server.cpython-314.pyc b/bird_cnn/__pycache__/server.cpython-314.pyc index ab31532..0cd2d06 100644 Binary files a/bird_cnn/__pycache__/server.cpython-314.pyc and b/bird_cnn/__pycache__/server.cpython-314.pyc differ diff --git a/bird_cnn/bird_cnn.py b/bird_cnn/bird_cnn.py index 580e4a6..b8cc332 100644 --- a/bird_cnn/bird_cnn.py +++ b/bird_cnn/bird_cnn.py @@ -50,7 +50,7 @@ class Bird_CNN(nn.Module): SkipBlock(c_in=c_hidden, c_out=c_hidden), SkipBlock(c_in=c_hidden, c_out=c_hidden), SkipBlock(c_in=c_hidden, c_out=c_hidden), - + SkipBlock(c_in=c_hidden, c_out=c_hidden*2), SkipBlock(c_in=c_hidden*2, c_out=c_hidden*2), SkipBlock(c_in=c_hidden*2, c_out=c_hidden*2), diff --git a/bird_cnn/build_models/bird_cnn b/bird_cnn/build_models/bird_cnn index 1fc5666..0d1ca27 100644 Binary files a/bird_cnn/build_models/bird_cnn and b/bird_cnn/build_models/bird_cnn differ diff --git a/bird_cnn/server.py b/bird_cnn/server.py index e6bd80c..96fa479 100644 --- a/bird_cnn/server.py +++ b/bird_cnn/server.py @@ -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)