search

imbue-ai / self_supervised

A Pytorch-Lightning implementation of self-supervised algorithms
MIT License
536 stars 52 forks source link

Deprecation warning #6

Closed JosephDiPalma closed 3 years ago

JosephDiPalma commented 3 years ago

Line 109 in lars.py has a UserWarning which says 

This overload of add_ is deprecated.
    add_(Number alpha, Tensor other)
Consider using one of the following signatures instead:
    add_(Tensor other, *, Number alpha) (Triggered internally at /pytorch/torch/csrc/utils/python_arg_parser.cpp:882.)
  buf.mul_(momentum).add_(actual_lr, d_p + weight_decay * p.data)

I'm using PyTorch 1.7.0 on CUDA 11.0.

abefetterman commented 3 years ago

Thanks for the heads up! Moved to the new signature for the add_ on that line.