wassname/DeepTime
1
0
Fork 0
mirror of https://github.com/wassname/DeepTime.git synced 2026-07-19 11:19:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

seq_len and other fixes

Browse Source
This commit is contained in:
wassname
2022-11-23 12:02:22 +08:00
parent fc1a605b01
commit cb27d2ded3
10 changed files with 488 additions and 374 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/modules/inr.py
+2 -2
View File
@@ -31,7 +31,7 @@ class INRLayer(nn.Module):
class INR(nn.Module):
def __init__(self, in_feats: int, layers: int, layer_size: int, n_fourier_feats: int, scales: float,
def __init__(self, in_feats: int, out_feats:int, layers: int, layer_size: int, n_fourier_feats: int, scales: float,
dropout: Optional[float] = 0.1):
super().__init__()
self.features = nn.Linear(in_feats, layer_size) if n_fourier_feats == 0 \
@@ -39,7 +39,7 @@ class INR(nn.Module):
in_size = layer_size if n_fourier_feats == 0 \
else n_fourier_feats
layers = [INRLayer(in_size, layer_size, dropout=dropout)] + \
[INRLayer(layer_size, layer_size, dropout=dropout) for _ in range(layers - 1)]
[INRLayer(layer_size, out_feats, dropout=dropout) for _ in range(layers - 1)]
self.layers = nn.Sequential(*layers)
def forward(self, x: Tensor) -> Tensor: