Typo

erikwijmans · erikwijmans · commit f4c3342c42a5 · 2018-09-14T09:52:58.000-04:00
diff --git a/train_sem_seg.py b/train_sem_seg.py
@@ -107,8 +107,8 @@
         model.parameters(), lr=args.lr, weight_decay=args.weight_decay
     )
 
-    lr_lbmd = lambda e: max(args.lr_decay**(int(it * args.batch_size / args.decay_step)), lr_clip / args.lr)
-    bnm_lmbd = lambda e: max(args.bn_momentum * args.bn_decay**(int(it * args.batch_size / args.decay_step)), bnm_clip)
+    lr_lbmd = lambda it: max(args.lr_decay**(int(it * args.batch_size / args.decay_step)), lr_clip / args.lr)
+    bnm_lmbd = lambda it: max(args.bn_momentum * args.bn_decay**(int(it * args.batch_size / args.decay_step)), bnm_clip)
 
     if args.checkpoint is None:
         lr_scheduler = lr_sched.LambdaLR(optimizer, lr_lbmd)

Original file line number	Diff line number	Diff line change
`@@ -107,8 +107,8 @@`
`107`	`107`	`model.parameters(), lr=args.lr, weight_decay=args.weight_decay`
`108`	`108`	`)`
`109`	`109`
`110`		`- lr_lbmd = lambda e: max(args.lr_decay*(int(it args.batch_size / args.decay_step)), lr_clip / args.lr)`
`111`		`- bnm_lmbd = lambda e: max(args.bn_momentum * args.bn_decay*(int(it args.batch_size / args.decay_step)), bnm_clip)`
	`110`	`+ lr_lbmd = lambda it: max(args.lr_decay*(int(it args.batch_size / args.decay_step)), lr_clip / args.lr)`
	`111`	`+ bnm_lmbd = lambda it: max(args.bn_momentum * args.bn_decay*(int(it args.batch_size / args.decay_step)), bnm_clip)`
`112`	`112`
`113`	`113`	`if args.checkpoint is None:`
`114`	`114`	`lr_scheduler = lr_sched.LambdaLR(optimizer, lr_lbmd)`