修复BUG

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

View File

@ -94,6 +94,8 @@ 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()
if cnt == 0:
onnx_path = str(battle.challenger_path) + '.onnx'
if os.path.exists(onnx_path):
os.remove(onnx_path)