Commit 6e918f1

author

Tree

committed

Update

1 parent 42f30f7 commit 6e918f1Copy full SHA for 6e918f1

File tree

+12

-11

lines changed

+12

-11

lines changed

Lines changed: 12 additions & 11 deletions

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,8 @@ def run():`
`39`	`39`	`)`
`40`	`40`
`41`	`41`	`# Training`
`42`		`- for code_length in [16, 32, 48, 128]:`
	`42`	`+ #for code_length in [16, 32, 48, 128]:`
	`43`	`+ for code_length in [16]:`
`43`	`44`	`args.code_length = code_length`
`44`	`45`	`checkpoint = hashnet.train(`
`45`	`46`	`train_dataloader,`
`@@ -57,16 +58,16 @@ def run():`
`57`	`58`	`logger.info('[code_length:{}][map:{:.4f}]'.format(args.code_length, checkpoint['map']))`
`58`	`59`
`59`	`60`	`# Save checkpoint`
`60`		`- torch.save(`
`61`		`- checkpoint,`
`62`		`- os.path.join('checkpoints', '{}_model_{}_code_{}_alpha_{}_map_{:.4f}.pt'.format(`
`63`		`- args.dataset,`
`64`		`- args.arch,`
`65`		`- args.code_length,`
`66`		`- args.alpha,`
`67`		`- checkpoint['map']),`
`68`		`- )`
`69`		`- )`
	`61`	`+ #torch.save(`
	`62`	`+ # checkpoint,`
	`63`	`+ # os.path.join('checkpoints', '{}_model_{}_code_{}_alpha_{}_map_{:.4f}.pt'.format(`
	`64`	`+ # args.dataset,`
	`65`	`+ # args.arch,`
	`66`	`+ # args.code_length,`
	`67`	`+ # args.alpha,`
	`68`	`+ # checkpoint['map']),`
	`69`	`+ # )`
	`70`	`+ #)`
`70`	`71`
`71`	`72`
`72`	`73`	`def load_config():`

Comments

(0)