Fixed crps + more inputs

2023-12-05 00:08:17 +00:00
parent 120b6aa5bd
commit d3bf04d68c
13 changed files with 128426 additions and 70 deletions
--- a/src/models/time_embedding_layer.py
+++ b/src/models/time_embedding_layer.py
@@ -5,7 +5,6 @@ class TimeEmbedding(nn.Module):
    def __init__(self, time_features: int, embedding_dim: int):
        super().__init__()
        self.time_features = time_features
-        print(time_features)
        self.embedding = nn.Embedding(time_features, embedding_dim)

    def forward(self, x):