Skip to content

Commit

Permalink
fix some bugs
Browse files Browse the repository at this point in the history
  • Loading branch information
caozhou committed May 28, 2024
1 parent 98ef573 commit 373175b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion flagscale/auto_tuner/prune/history.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ def prune_by_micro_batch_size(config, strategy, history=[]):
logger.info(
f"The strategy {strategy} has been pruned by micro_batch_size performance."
)
strategy["performance"] = performance
strategy["performance"] = item["performance"]
strategy["max_mem"] = item["max_mem"]
strategy["pruned"] = True
return True
Expand Down

0 comments on commit 373175b

Please sign in to comment.