/aosp_15_r20/external/pytorch/torch/optim/ |
H A D | lr_scheduler.py | 95 self, optimizer: Optimizer, last_epoch=-1, verbose="deprecated" argument 322 last_epoch=-1, argument 422 last_epoch=-1, argument 527 last_epoch=-1, argument 586 last_epoch=-1, argument 653 last_epoch=-1, argument 732 last_epoch=-1, argument 806 self, optimizer: Optimizer, gamma: float, last_epoch=-1, verbose="deprecated" argument 862 last_epoch=-1, argument 997 last_epoch=-1, argument [all …]
|
H A D | swa_utils.py | 407 last_epoch=-1, argument
|
/aosp_15_r20/external/pytorch/torch/ao/pruning/scheduler/ |
H A D | lambda_scheduler.py | 33 def __init__(self, sparsifier, sl_lambda, last_epoch=-1, verbose=False): argument
|
H A D | cubic_scheduler.py | 50 last_epoch=-1, argument
|
H A D | base_scheduler.py | 14 def __init__(self, sparsifier, last_epoch=-1, verbose=False): argument
|
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/optim/schedulers/ |
H A D | reduce_on_plateau_scheduler.h | 60 int last_epoch; variable
|
/aosp_15_r20/external/pytorch/test/ao/sparsity/ |
H A D | test_data_scheduler.py | 21 def __init__(self, sparsifier, sparsifier_hyperparam, last_epoch=-1, verbose=False): argument
|
/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_scheduler/ |
H A D | base_data_scheduler.py | 34 self, data_sparsifier, schedule_param: str, last_epoch=-1, verbose=False argument
|
/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/lightning/tests/ |
H A D | test_callbacks.py | 65 last_epoch=-1, argument
|
/aosp_15_r20/external/pytorch/test/optim/ |
H A D | test_lrscheduler.py | 97 def __init__(self, optimizer, gamma, milestones, last_epoch=-1): argument
|