Update tools.py · 5da8bd07e2 - Prediction-MindSpore - 开源协同云脑生态支撑系统

Update tools.py

main

BBing 2 years ago

parent

89b4102a44

commit

5da8bd07e2

1 changed files with 1 additions and 0 deletions

Unified View

							
								+ 1
								
								- 0
							
utils/tools.pyView File
								
																@@ -3,6 +3,7 @@ import mindspore.numpy as mnp
															
																import mindspore
																
																import mindspore
															
																from mindspore import Tensor, Parameter
																
																from mindspore import Tensor, Parameter
															
																def adjust_learning_rate(optimizer, epoch, args):
																
																def adjust_learning_rate(optimizer, epoch, args):
															
																    if args.lradj == 'type1':
																
																    if args.lradj == 'type1':
															
																        lr_adjust = {epoch: args.learning_rate * (0.5 ** ((epoch-1) // 1))}
																
																        lr_adjust = {epoch: args.learning_rate * (0.5 ** ((epoch-1) // 1))}