Team Ai
Datasetpublic

SparseWake/sparsewake

SparseWake SparseWake is a synthetic benchmark for sparse temporal hydrodynamic sensing. ICLR 2027 release The expanded release adds controlled multi-source mixtures and common-prior nearest-source tasks, with complete core data banks, reference checkpoints, a small review supplement, and reproduction code with a frozen wake-library input. Download release iclr2027-v1.0rc2 The version page lists the three archives, exact sizes, checksums, extraction instructions… See the full description on the dataset page: https://huggingface.co/datasets/SparseWake/sparsewake.

sourceHugging Facecc-by-4.0updated 14d agoView on Hugging Face
0likes279downloads
models.py21 linesDownload Raw Back to sparsewake
1from __future__ import annotations2 3import torch4from torch import nn5 6 7class TemporalMLP(nn.Module):8    def __init__(self, input_dim: int, output_dim: int = 2, hidden: tuple[int, ...] = (256, 256, 128), dropout: float = 0.05):9        super().__init__()10        layers: list[nn.Module] = []11        last = input_dim12        for width in hidden:13            layers.extend([nn.Linear(last, width), nn.LayerNorm(width), nn.GELU(), nn.Dropout(dropout)])14            last = width15        layers.append(nn.Linear(last, output_dim))16        self.net = nn.Sequential(*layers)17 18    def forward(self, x: torch.Tensor) -> torch.Tensor:19        return self.net(x)20 21