Did you also call model.module.feature during training? It seems to be a custom method, so that I think nn.DataParallel won’t be applied.
Also, wrap your code in a with torch no_grad() block to save some memory during evaluation.