交通轨迹预测调度模块

调度模块参数

• gpu：是否是用gpu训练，默认True。

• gpu_id：指定使用的gpu的id，默认0。

• learning_rate: 学习率，默认0.0005。

• L2: torch优化器的L2范数，默认0.00001。

• max_epoch: 训练最大轮数，默认1。

• lr_step: torch学习率下降的轮数，即经过lr_step轮后还没改进时则下降学习率，默认2。

• lr_decay: torch每次学习率下降的比例，默认0.1。

• clip: 梯度裁剪中，梯度的最大范数，默认5.0。

• schedule_threshold: torch scheduler的门限值，默认0.001。

• verbose: 输出结果前的训练次数，默认10。