修复BUG
This commit is contained in:
parent
4f8db44ae4
commit
0e3476b734
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue