fgnt / padertorch

A collection of common functionality to simplify the design, training and evaluation of machine learning models based on pytorch with an emphasis on speech processing.
MIT License
71 stars 16 forks source link

Add workaround for torch==1.12.0 bug #151

Closed alexanderwerning closed 1 year ago

alexanderwerning commented 1 year ago

Resuming from a checkpoint in torch==1.12.0 is broken, this was fixed in torch=1.12.1. This workaround allows to load checkpoints with version 1.12.0 as well. In https://github.com/pytorch/pytorch/issues/80809 a 10% slowdown was reported, which I did not observe.