在python程序被终止的“弥留之际”再执行最后一段代码

对于深度学习训练或者验证过程，由于数据太多，很多时候我们等不及程序执行完毕就直接“Control+C”终止了程序运行。但是，我们还想程序在结束运行之前，清理一下环境，删除一些文件。于是，我们可以用下面的方法完成，非常简单。无论按了多少次“Control+C”，都不用担心，下面的代码都能成功执行。

import atexit
import glob

@atexit.register
def clean():
    print('-'*100)
    print('clean up the environment ...')
    path = 'ResNet_Tensor/'
    for infile in glob.glob(os.path.join(path, '*.pt')):
        os.remove(infile)
    print('Done')