From 0e3476b734eeaf3c572e8bfe2ec6526fbd099475 Mon Sep 17 00:00:00 2001 From: zhiyang7 Date: Wed, 5 Jan 2022 12:05:07 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DBUG?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- douzero/server/battle.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/douzero/server/battle.py b/douzero/server/battle.py index 8222763..df9bd0e 100644 --- a/douzero/server/battle.py +++ b/douzero/server/battle.py @@ -94,10 +94,12 @@ def battle_logic(flags, baseline : Baseline, battle : Battle): battle.status = 1 if challenge_success else 2 battle.save() if not challenge_success: - onnx_path = str(battle.challenger_path) + '.onnx' - if os.path.exists(onnx_path): - os.remove(onnx_path) - os.remove(str(battle.challenger_path)) + cnt = Battle.select().where(Battle.status == 0).count() + if cnt == 0: + onnx_path = str(battle.challenger_path) + '.onnx' + if os.path.exists(onnx_path): + os.remove(onnx_path) + os.remove(str(battle.challenger_path)) else: baseline_players[position_idx[battle.challenger_position]] = DeepAgent(battle.challenger_position, str(battle.challenger_path), use_onnx=True)