popgym.baselines.models.indrnn

Classes

Module Contents

class popgym.baselines.models.indrnn.IndRNN(input_size, hidden_size, activation='relu', clamp=True, max_len=1024)

Bases: torch.nn.Module

clamp = True

clamp_val = 1.0006771306930664

input0

weight0

bias0

input1

weight1

bias1

output

clamp_weights() → None

forward(x: torch.Tensor, state: torch.Tensor | None = None) → Tuple[torch.Tensor, torch.Tensor]

input:: x: [B, T, F] state: [B, 2, F]
output:: y: [B, T, D] state: [B, 2, D]