修复BUG

This commit is contained in:
zhiyang7 2022-01-05 12:06:32 +08:00
parent 0e3476b734
commit 4f8cf8266d
1 changed files with 1 additions and 1 deletions

View File

@ -94,7 +94,7 @@ def battle_logic(flags, baseline : Baseline, battle : Battle):
battle.status = 1 if challenge_success else 2
battle.save()
if not challenge_success:
cnt = Battle.select().where(Battle.status == 0).count()
cnt = Battle.select().where(Battle.challenger_path == battle.challenger_path,Battle.status == 0).count()
if cnt == 0:
onnx_path = str(battle.challenger_path) + '.onnx'
if os.path.exists(onnx_path):