terminate called after throwing an instance of 'std::bad_alloc'

[panlanlan]

when i run the train.py
terminate called after throwing an instance of 'std::bad_alloc'
what(): std::bad_alloc
i run the code on the cpu　ubuntu18.04
can you help me solve the problem