@@ -325,7 +325,7 @@ public class PolynomialLR : LRScheduler
325325 /// <param name="last_epoch">The index of last epoch. Default: -1.</param>
326326 /// <param name="verbose"> If true, prints a message to stdout for each update. Default: false.</param>
327327 /// <returns>A scheduler</returns>
328- public PolynomialLR ( Optimizer optimizer , int total_iters = 5 , int power = 1 , int last_epoch = - 1 , bool verbose = false ) : base ( optimizer , last_epoch , verbose )
328+ public PolynomialLR ( Optimizer optimizer , int total_iters = 5 , double power = 1.0 , int last_epoch = - 1 , bool verbose = false ) : base ( optimizer , last_epoch , verbose )
329329 {
330330 if ( optimizer == null ) throw new ArgumentNullException ( "optimizer" ) ;
331331 _power = power ;
@@ -359,7 +359,7 @@ protected override IEnumerable<double> get_closed_form_lr()
359359 }
360360
361361 private double _total_iters ;
362- private int _power ;
362+ private double _power ;
363363 }
364364
365365 /// <summary>
@@ -1306,7 +1306,7 @@ public static LRScheduler MultiStepLR(Optimizer optimizer, IList<int> milestones
13061306 /// <param name="last_epoch">The index of last epoch. Default: -1.</param>
13071307 /// <param name="verbose"> If true, prints a message to stdout for each update. Default: false.</param>
13081308 /// <returns>A scheduler</returns>
1309- public static LRScheduler PolynomialLR ( Optimizer optimizer , int total_iters = 5 , int power = 1 , int last_epoch = - 1 , bool verbose = false )
1309+ public static LRScheduler PolynomialLR ( Optimizer optimizer , int total_iters = 5 , double power = 1 , int last_epoch = - 1 , bool verbose = false )
13101310 {
13111311 return new impl . PolynomialLR ( optimizer , total_iters , power , last_epoch , verbose ) ;
13121312 }
@@ -1398,7 +1398,7 @@ public static LRScheduler SequentialLR(Optimizer optimizer, IEnumerable<LRSchedu
13981398 /// </param>
13991399 /// <param name="verbose">If true, prints a message to stdout for each update. Default: false.</param>
14001400 /// <returns>A scheduler</returns>
1401- public static LRScheduler LinearLR ( Optimizer optimizer , double start_factor = 1.0 / 3 , double end_factor = 5 , int total_iters = 5 , int last_epoch = - 1 , bool verbose = false )
1401+ public static LRScheduler LinearLR ( Optimizer optimizer , double start_factor = 1.0 / 3 , double end_factor = 1.0 , int total_iters = 5 , int last_epoch = - 1 , bool verbose = false )
14021402 {
14031403 return new impl . LinearLR ( optimizer , start_factor , end_factor , total_iters , last_epoch , verbose ) ;
14041404 }
0 commit comments