Add memory demanding to pytorch models

4 jobs for feature-extractors in 35 minutes and 7 seconds