pytorch-warmup · PyPI
https://pypi.org/project/pytorch-warmup/
WEBOct 24, 2022 · Approach 1. When the learning rate schedule uses the global iteration number, the untuned linear warmup can be used as follows: import torch import pytorch_warmup as warmup optimizer = torch.optim.AdamW(params, lr=0.001, betas=(0.9, 0.999), weight_decay=0.01) num_steps = len(dataloader) * num_epochs …
DA: 50 PA: 42 MOZ Rank: 99