Update for Rebuttal

2020-12-01 12:34:00 +08:00
parent 29428bf5a3
commit 8afb62ad2e
5 changed files with 96 additions and 6 deletions
--- a/lib/nats_bench/api_topology.py
+++ b/lib/nats_bench/api_topology.py
@@ -83,6 +83,7 @@ class NATStopology(NASBenchMetaAPI):
    self._search_space_name = 'topology'
    self._fast_mode = fast_mode
    self._archive_dir = None
+    self._full_train_epochs = 200
    self.reset_time()
    if file_path_or_dict is None:
      if self._fast_mode: