Skip to content

Commit

Permalink
add comment
Browse files Browse the repository at this point in the history
  • Loading branch information
namsaraeva committed May 17, 2024
1 parent b93cb05 commit 8281331
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/sparcscore/ml/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ def make_layers_MLP(self, cfg_MLP, cfg, regression = False):
# get output feature size of CNN with chosen configuration

if regression:
in_features = int(cfg[-2]) * 4 * 4
in_features = int(cfg[-2]) * 4 * 4 # needed for the STING screen since the images are big
else:
in_features = int(cfg[-2]) * 2 * 2

Expand Down

0 comments on commit 8281331

Please sign in to comment.