Unverified Commit 6427f50a by Pan Yinxu Committed by GitHub

delete redundant append_to_dict (#236)

Closes #227
parent d0725a62
...@@ -364,9 +364,6 @@ class MegatronPPOActor(BasePPOActor): ...@@ -364,9 +364,6 @@ class MegatronPPOActor(BasePPOActor):
else: else:
raise NotImplementedError raise NotImplementedError
for metric in metric_micro_batch:
append_to_dict(metrics, metric) # append the metric from this micro-batch to global metrics.
# add empty cache after each compute # add empty cache after each compute
torch.cuda.empty_cache() torch.cuda.empty_cache()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment