Update for Rebuttal

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:

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:

lib/nats_bench/api_utils.py

@@ -190,6 +190,10 @@ class NASBenchMetaAPI(metaclass=abc.ABCMeta):
   def archive_dir(self):
     return self._archive_dir
 
+  @property
+  def full_train_epochs(self):
+    return self._full_train_epochs
+
   def reset_archive_dir(self, archive_dir):
     self._archive_dir = archive_dir