[python]捕获ctrl+C事件后处理

Posted 2022-08-02 FL1623863129

使用try方法：

	try:
		model.learn(total_timesteps=int(trian_steps), callback=callback)
	except KeyboardInterrupt:
		model.save(log_dir + "--".format(algo, env_name, trian_steps))
		results_plotter.plot_results([log_dir], trian_steps, results_plotter.X_TIMESTEPS, ", ".format(algo, env_name))
		plt.savefig(log_dir + "--.png".format(algo, env_name, trian_steps))
		# plt.show()
		print('  training finished.'.format(algo, env))

使用signal模块

import signal

def exit(signum, frame):
  print('You choose to stop me.')
  exit()

signal.signal(signal.SIGINT, exit)
signal.signal(signal.SIGTERM, exit)

while 1:
  print('running......)