Exploding memory usage on TimeDistributedResNet

broadinstitute / keras-resnet

Keras package for deep residual networks

Other

300 stars 127 forks source link

Exploding memory usage on TimeDistributedResNet #38

Open oxrider opened 6 years ago

oxrider commented 6 years ago

I'm trying to use TimeDistributedResNet50 in my project and the memory usage explodes (used up 64GB RAM and 35+GB Swap memory before my computer dies) with a sequence length of only 3. Can you confirm if this is expected?

I tried TimeDistributed(VGG16) and the memory usage is only 1.9GB at sequence length 20.