@@ -325,7 +325,7 @@ public class PolynomialLR : LRScheduler
325
325
/// <param name="last_epoch">The index of last epoch. Default: -1.</param>
326
326
/// <param name="verbose"> If true, prints a message to stdout for each update. Default: false.</param>
327
327
/// <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 )
329
329
{
330
330
if ( optimizer == null ) throw new ArgumentNullException ( "optimizer" ) ;
331
331
_power = power ;
@@ -359,7 +359,7 @@ protected override IEnumerable<double> get_closed_form_lr()
359
359
}
360
360
361
361
private double _total_iters ;
362
- private int _power ;
362
+ private double _power ;
363
363
}
364
364
365
365
/// <summary>
@@ -1306,7 +1306,7 @@ public static LRScheduler MultiStepLR(Optimizer optimizer, IList<int> milestones
1306
1306
/// <param name="last_epoch">The index of last epoch. Default: -1.</param>
1307
1307
/// <param name="verbose"> If true, prints a message to stdout for each update. Default: false.</param>
1308
1308
/// <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 )
1310
1310
{
1311
1311
return new impl . PolynomialLR ( optimizer , total_iters , power , last_epoch , verbose ) ;
1312
1312
}
@@ -1398,7 +1398,7 @@ public static LRScheduler SequentialLR(Optimizer optimizer, IEnumerable<LRSchedu
1398
1398
/// </param>
1399
1399
/// <param name="verbose">If true, prints a message to stdout for each update. Default: false.</param>
1400
1400
/// <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 )
1402
1402
{
1403
1403
return new impl . LinearLR ( optimizer , start_factor , end_factor , total_iters , last_epoch , verbose ) ;
1404
1404
}
0 commit comments