Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:eelregit/map2map

Browse Source 
Conflicts:
	map2map/train.py
This commit is contained in:
Yin Li 2019-12-03 18:07:11 -05:00
commit 34b8c62c76
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/train.py
View File

@ -90,7 +90,7 @@ def gpu_worker(local_rank, args):
)
#scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer)
scheduler = torch.optim.lr_scheduler.CyclicLR(optimizer,
base_lr=args.lr * 1e-2, max_lr=args.lr)
base_lr=args.lr * 1e-2, max_lr=args.lr, cycle_momentum=False)
if args.load_state:
state = torch.load(args.load_state, map_location=args.device)