Hey @ilyes, thanks for this solution. Is it possibile to “pack” inside TimeDistriubted also:
nn.ReLU()
nn.Dropout()

like:
TimeDistributed(nn.ReLU(inplace=True), time_steps = 20)
TimeDistributed(nn.Dropout(p=0.5), time_steps = 20)

Thanks.