search

cypw / PyTorch-MFNet

MIT License
252 stars 56 forks source link

Out of memory #25

Open buaa-luzhi opened 5 years ago

buaa-luzhi commented 5 years ago

@cypw Hello! Out of memory will occur during the training and testing of ucf-101 data. I only have a 32G CPU. Is this a normal phenomenon? How much memory is needed to train the network properly? Thank you very much!

cypw commented 5 years ago

32 GB RAM on the CPU side should be enough.

You can try to reduce num_woker to save some RAM: https://github.com/cypw/PyTorch-MFNet/blob/master/data/iterator_factory.py#L183