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

Update tools.py

main

BBing 2 years ago

parent

fdeff5fe29

commit

8c562294dc

1 changed files with 1 additions and 1 deletions

Unified View

							
								+ 1
								
								- 1
							
utils/tools.pyView File
								
																@@ -1,6 +1,6 @@
															
																import numpy as np
																
																import numpy as np
															
																import torch
																
																import torch
															
																1
															
																def adjust_learning_rate(optimizer, epoch, args):
																
																def adjust_learning_rate(optimizer, epoch, args):
															
																    # lr = args.learning_rate * (0.2 ** (epoch // 2))
																
																    # lr = args.learning_rate * (0.2 ** (epoch // 2))
															
																    if args.lradj=='type1':
																
																    if args.lradj=='type1':