Home
last modified time | relevance | path

Searched defs:last_epoch (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/optim/
H A Dlr_scheduler.py95 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 Dswa_utils.py407 last_epoch=-1, argument
/aosp_15_r20/external/pytorch/torch/ao/pruning/scheduler/
H A Dlambda_scheduler.py33 def __init__(self, sparsifier, sl_lambda, last_epoch=-1, verbose=False): argument
H A Dcubic_scheduler.py50 last_epoch=-1, argument
H A Dbase_scheduler.py14 def __init__(self, sparsifier, last_epoch=-1, verbose=False): argument
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/optim/schedulers/
H A Dreduce_on_plateau_scheduler.h60 int last_epoch; variable
/aosp_15_r20/external/pytorch/test/ao/sparsity/
H A Dtest_data_scheduler.py21 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 Dbase_data_scheduler.py34 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 Dtest_callbacks.py65 last_epoch=-1, argument
/aosp_15_r20/external/pytorch/test/optim/
H A Dtest_lrscheduler.py97 def __init__(self, optimizer, gamma, milestones, last_epoch=-1): argument