From d7f0756504a4c9d7ad892d4ea58d0c7c11e4fd08 Mon Sep 17 00:00:00 2001 From: namsaraeva Date: Thu, 16 May 2024 13:56:17 +0200 Subject: [PATCH] printmsg --- src/sparcscore/ml/models.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/sparcscore/ml/models.py b/src/sparcscore/ml/models.py index 7c449f3..b7a7f5b 100644 --- a/src/sparcscore/ml/models.py +++ b/src/sparcscore/ml/models.py @@ -172,9 +172,7 @@ def __init__(self, ): super(VGG2_regression, self).__init__() - - self.norm = nn.BatchNorm2d(in_channels) - + self.norm = nn.BatchNorm2d(in_channels) self.features = self.make_layers(self.cfgs[cfg], in_channels) self.classifier = self.make_layers_MLP(self.cfgs_MLP[cfg_MLP], self.cfgs[cfg], regression=True) # regression is set to True to make the final layer a single output @@ -183,6 +181,7 @@ def vgg(cfg, in_channels, **kwargs): return model def forward(self, x): + print(x.shape) x = self.norm(x) x = self.features(x)