stubs/torch/optim/optimizer.pyi · fb49b515dbbec7c859c9f1231d6d904558cac774 · OpenDAS / fairscale

[fix] optim/oss: fix state cast (#56) · fb49b515

msbaines authored Aug 27, 2020

Workaround PyTorch bug that casts state (pytorch/pytorch#43706).

Copied from https://github.com/pytorch/fairseq/blob/v0.9.0/fairseq/optim/fp16_optimizer.py#L251-L268

fb49b515

optimizer.pyi 641 Bytes

Replace optimizer.pyi