iicd/naswot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed CIFAR-10 validation splitting

Browse Source
This commit is contained in:
jack-willturner 2020-06-03 14:21:49 +01:00
parent 357e877e8d
commit e5d7a944c9
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
search.py
View File

@ -39,6 +39,7 @@ torch.manual_seed(args.seed)
import torchvision.transforms as transforms import torchvision.transforms as transforms
from datasets import get_datasets from datasets import get_datasets
from config_utils import load_config
from nas_201_api import NASBench201API as API from nas_201_api import NASBench201API as API
def get_batch_jacobian(net, x, target, to, device, args=None): def get_batch_jacobian(net, x, target, to, device, args=None):