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_size.py
+++ b/lib/nats_bench/api_size.py
@@ -83,6 +83,7 @@ class NATSsize(NASBenchMetaAPI):
    self._search_space_name = 'size'
    self._fast_mode = fast_mode
    self._archive_dir = None
+    self._full_train_epochs = 90
    self.reset_time()
    if file_path_or_dict is None:
      if self._fast_mode: