Update pretrain_IJEPA.py by fixing a UserWarning in add_

gaasher / I-JEPA

Implementation of I-JEPA from "Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture"

MIT License

249 stars 24 forks source link

Update pretrain_IJEPA.py by fixing a UserWarning in add_ #2

Closed AymenSe closed 1 year ago

AymenSe commented 1 year ago

Fixing the warning: "UserWarning: 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 /opt/conda/conda-bld/pytorch_1595661244635/work/torch/csrc/utils/python_arg_parser.cpp:797.)