Skip to content

Commit

Permalink
remove TODO
Browse files Browse the repository at this point in the history
  • Loading branch information
LarsKue committed Dec 26, 2023
1 parent fc2f2ce commit 0d1b642
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/models/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,8 +100,6 @@ def __init__(self, hparams, *datasets):

# medium networks (~15M params)

# TODO: check if dropout necessary

self.encoder = nn.Sequential(
nn.Sequential(nn.Linear(self.hparams.features, 64), nn.ReLU()),
*[nn.Sequential(SkipLinear(64), nn.ReLU()) for _ in range(4)],
Expand Down

0 comments on commit 0d1b642

Please sign in to comment.