sane error message after MemoryError in train.py