Sync NATS-Bench's v1.0 and update algorithm names

2020-10-15 21:56:10 +11:00
parent 10e5f05935
commit 7d55192d83
7 changed files with 28 additions and 26 deletions
--- a/lib/nats_bench/api_utils.py
+++ b/lib/nats_bench/api_utils.py
@@ -939,9 +939,9 @@ class ArchResults(object):
    x.load_state_dict(state_dict)
    return x

-  # This function is used to clear the weights saved in each 'result'
-  # This can help reduce the memory footprint.
  def clear_params(self):
+    """Clear the weights saved in each 'result'."""
+    # NOTE(xuanyidong): This can help reduce the memory footprint.
    for unused_key, result in self.all_results.items():
      del result.net_state_dict
      result.net_state_dict = None