Skip to content

Commit 4a62b57

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent c96d440 commit 4a62b57

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

neural_network/real_time_encoder_transformer.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Time2Vec layer for positional encoding of real-time data like EEG
88
class Time2Vec(nn.Module):
9-
# Encodes time steps into a continuous embedding space
9+
# Encodes time steps into a continuous embedding space
1010
def __init__(self, d_model):
1111
super().__init__()
1212
self.w0 = nn.Parameter(torch.randn(1, 1))
@@ -174,12 +174,8 @@ def __init__(
174174

175175
# Transformer encoder for sequence modeling
176176
self.encoder = TransformerEncoder(
177-
d_model,
178-
n_head,
179-
hidden_dim,
180-
num_layers,
181-
drop_prob
182-
)
177+
d_model, n_head, hidden_dim, num_layers, drop_prob
178+
)
183179

184180
# Attention pooling to summarize time dimension
185181
self.pooling = AttentionPooling(d_model)

0 commit comments

Comments
 (0)