Skip to content

Commit

Permalink
printmsg
Browse files Browse the repository at this point in the history
  • Loading branch information
namsaraeva committed May 16, 2024
1 parent 3e5ace5 commit d7f0756
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/sparcscore/ml/models.py
Original file line number Diff line number Diff line change
@@ -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)

0 comments on commit d7f0756

Please sign in to comment.