pytorch优化器lr调度程序
torch-lr-scheduler的Python项目详细描述
火炬lr调度程序
pytorch优化器lr调度程序。
安装
需要Python3.6+。
pip install torch-lr-scheduler
用法
importtorch_lr_schedulerlr_scheduler=torch_lr_scheduler.factory(config={'line_chain':[{# warm up to 0.8'mode':'linear','ratio':0.01,'start':0.2,'target':0.8},{# cosine to 0.0'mode':'cosine','ratio':1.0,'target':0.0}]})print(lr_scheduler)#> LrScheduler (learning_rate_scale: 1.0) with LineChain (#> 1.0%, linear from 0.2 to 0.8,#> 100.0%, cosine from 0.8 to 0.0,#> )