0
点赞
收藏
分享

微信扫一扫

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


使用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......)

举报

相关推荐

0 条评论