integrate ground truth (json file) of NASBenchMB

hyperbox/networks/nasbench_mbnet/network.py

@@ -1,5 +1,6 @@
 import torch.nn as nn
 import torch
+import inspect
 from collections import OrderedDict
 
 from hyperbox.mutables.spaces import OperationSpace
@@ -92,6 +93,7 @@ def __init__(
             assert len(arch_list) == sum(stages)
         self.arch_list = arch_list
         self.arch_info = None
+        self.QUERY_FILE_PATH = inspect.getfile(self.__class__).replace('network.py', 'nasbench_mbnet_cifar10.json')
 
         self.stem = nn.Sequential(
             nn.Conv2d(3, init_channels, 3, padding=1, bias=False),
@@ -143,9 +145,11 @@ def arch(self):
                     arch_list.append(block.mask.cpu().detach().numpy().argmax())
         return ''.join([str(x) for x in arch_list])
 
-    def query_by_key(self, query_file_path, key='mean_acc', arch=None):
+    def query_by_key(self, query_file_path=None, key='mean_acc', arch=None):
         if arch is None:
             arch = self.arch
+        if query_file_path is None:
+            query_file_path = self.QUERY_FILE_PATH
         if self.arch_info is None:
             import json
             with open(query_file_path, 'r') as f: