ZhikangNiu / encodec-pytorch

unofficial implementation of the High Fidelity Neural Audio Compression

scheduler.step called multiple times?

leoauri opened this issue 2 months ago · comments

Leo Auri commented 2 months ago

Hi, I see scheduler.step() called in https://github.com/ZhikangNiu/encodec-pytorch/blob/main/train_multi_gpu.py#L65 (if config.common.amp) and in https://github.com/ZhikangNiu/encodec-pytorch/blob/main/train_multi_gpu.py#L99. Is this a bug?

Zhikang Niu commented 2 months ago

@leoauri amp training isn't work so don't use amp training

Leo Auri commented 2 months ago

Thanks for your prompt reply!