lucidrains / vector-quantize-pytorch

Vector (and Scalar) Quantization, in Pytorch
MIT License
2.65k stars 215 forks source link

There is a future warning #155

Closed Cyunaam-Wei closed 2 months ago

Cyunaam-Wei commented 3 months ago

anaconda3\envs\pytorch39\lib\site-packages\vector_quantize_pytorch\vector_quantize_pytorch.py:461: FutureWarning: torch.cuda.amp.autocast(args...) is deprecated. Please use torch.amp.autocast('cuda', args...) instead.

hide-on-bush-lalala commented 2 months ago

碰到了一样的情况,肯呢个需要作者替换掉这个API,出现了大量的amp的FutureWarning。

hide-on-bush-lalala commented 2 months ago

我也可以尝试去修改这部分代码,但是对于github合作以及PR的模式不是很懂,有没有什么入门的演示。。。

Cyunaam-Wei commented 2 months ago

我也可以尝试去修改这部分代码,但是对于github合作以及PR的模式不是很懂,有没有什么入门的演示。。。

如果作者不更新的话,只能先忽略警告了