trying to convert keras code to pytorch

[InnovArul (Arul)]

Something like this might work for your case:

Timedistributed CNN

Hi there, I’m not sure that using for-loop will save the memory… I think those two will have the same memory cost in training. I wrote one for myself too… class TimeDistributed(nn.Module): def __init__(self, module, batch_first=False): super(TimeDistributed, self).__init__() self.module = module self.batch_first = batch_first def forward(self, x): ''' x size: (batch_size, time_steps, in_channels, height, width) ''' batch_size, time_steps, C, H,…