benchmarks.hota.learned_modules.Fuser.gate

gate = nn.Sequential(nn.Linear(2 * dim, hidden), nn.Tanh(), nn.Linear(hidden, dim), nn.Sigmoid())