From b0aa64847f59e8267d7df1807d41312eebfba41e Mon Sep 17 00:00:00 2001 From: cortexmancer Date: Wed, 4 Feb 2026 16:37:47 -0300 Subject: [PATCH] chore: correct value for linear0 layer --- breaching/cases/models/model_preparation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/breaching/cases/models/model_preparation.py b/breaching/cases/models/model_preparation.py index 0d87320d..32d0b8f2 100644 --- a/breaching/cases/models/model_preparation.py +++ b/breaching/cases/models/model_preparation.py @@ -352,7 +352,7 @@ def _construct_vision_model(cfg_model, cfg_data, pretrained=True, **kwargs): ("conv4", torch.nn.Conv2d(128, 256, 3, stride=1, padding=1)), ("relu3", torch.nn.LeakyReLU()), ("flatt", torch.nn.Flatten()), - ("linear0", torch.nn.Linear(12544, 12544)), + ("linear0", torch.nn.Linear(16384, 12544)), ("relu4", torch.nn.LeakyReLU()), ("linear1", torch.nn.Linear(12544, classes)), ("softmax", torch.nn.Softmax(dim=1)),