net = nn.Sequential(nn.Linear(dim, hidden), nn.Tanh(), nn.Linear(hidden, dim))
Source: unitrack/benchmarks/hota/learned_modules.py:32
