| 27 | |
| 28 | |
| 29 | class SchedulerType(Enum): |
| 30 | LINEAR = "linear" |
| 31 | COSINE = "cosine" |
| 32 | COSINE_WITH_RESTARTS = "cosine_with_restarts" |
| 33 | POLYNOMIAL = "polynomial" |
| 34 | CONSTANT = "constant" |
| 35 | CONSTANT_WITH_WARMUP = "constant_with_warmup" |
| 36 | PIECEWISE_CONSTANT = "piecewise_constant" |
| 37 | |
| 38 | |
| 39 | def get_constant_schedule(optimizer: Optimizer, last_epoch: int = -1) -> LambdaLR: |
no outgoing calls
no test coverage detected
searching dependent graphs…