Fixed CIFAR-10 validation splitting

2020-06-03 14:21:49 +01:00
parent 357e877e8d
commit e5d7a944c9
1 changed files with 1 additions and 0 deletions
--- a/search.py
+++ b/search.py
@@ -39,6 +39,7 @@ torch.manual_seed(args.seed)

 import torchvision.transforms as transforms
 from datasets import get_datasets
+from config_utils import load_config
 from nas_201_api import NASBench201API as API

 def get_batch_jacobian(net, x, target, to, device, args=None):