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

With table generator

Browse Source
This commit is contained in:
jack-willturner
2020-06-03 15:59:48 +01:00
parent 6f97d1be37
commit 45901dd7ec
5 changed files with 104 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
search.py
View File

@@ -153,5 +153,5 @@ state = {'accs': acc,
}
dset = args.dataset if not args.trainval else 'cifar10-valid'
fname = f"{args.save_loc}/{dset}_{args.n_runs}_{args.n_samples}_{args.mc_samples}_{args.alpha}_{args.seed}.t7"
fname = f"{args.save_loc}/{dset}_{args.n_runs}_{args.n_samples}_{args.seed}.t7"
torch.save(state, fname)