使用gc.collect()进行垃圾回收

import gc
del train, train_label, train_x, train_y, val_x, val_y
gc.collect()

垃圾回收开始的时候当前所有线程都将被挂起，开始收集托管堆上的垃圾，收集完了还要压缩内存，然后等待垃圾回收结束以后再恢复这些线程，从这个角度来说，还是少调用垃圾回收，但是不是不能调，要视情况而定。 【占用内存较多】

参考：

1. Python使用gc模块进行垃圾回收：https://www.jianshu.com/p/b6a20c812ce4
2. GC.Collect()简介及一般用法：https://blog.csdn.net/weixin_30632883/article/details/97307112