recognition/partial_fc/mxnet/memory_scheduler.py

from default import config
import mxnet as mx


def get_scheduler():
    step = [int(x) for x in config.lr_steps.split(',')]
    backbone_lr_scheduler = mx.lr_scheduler.MultiFactorScheduler(
        step=step, factor=0.1, base_lr=config.backbone_lr)
    memory_bank_lr_scheduler = mx.lr_scheduler.MultiFactorScheduler(
        step=step, factor=0.1, base_lr=config.memory_bank_lr)

    return backbone_lr_scheduler, memory_bank_lr_scheduler
Update code 2020-10-13 13:13:24 +08:00			`from default import config`
			`import mxnet as mx`


			`def get_scheduler():`
			`step = [int(x) for x in config.lr_steps.split(',')]`
			`backbone_lr_scheduler = mx.lr_scheduler.MultiFactorScheduler(`
refine repo structure 2020-11-06 13:59:21 +08:00			`step=step, factor=0.1, base_lr=config.backbone_lr)`
Update code 2020-10-13 13:13:24 +08:00			`memory_bank_lr_scheduler = mx.lr_scheduler.MultiFactorScheduler(`
			`step=step, factor=0.1, base_lr=config.memory_bank_lr)`

			`return backbone_lr_scheduler, memory_bank_lr_scheduler`