search

automl / TabPFN

Official implementation of the TabPFN paper (https://arxiv.org/abs/2207.01848) and the tabpfn package.
http://priorlabs.ai
Apache License 2.0
1.22k stars 109 forks source link

UserWarning from torch.utils.checkpoint #88

Open DhanshreeA opened 7 months ago

DhanshreeA commented 7 months ago

I am running tabpfn v0.0.9 within a Python 3.10 environment, and torch version 2.2.1. There's the following user warning: 

/usr/local/lib/python3.10/dist-packages/torch/utils/checkpoint.py:460: UserWarning: torch.utils.checkpoint: please pass in use_reentrant=True or use_reentrant=False explicitly. The default value of use_reentrant will be updated to be False in the future. To maintain current behavior, pass use_reentrant=True. It is recommended that you use use_reentrant=False. Refer to docs for more details on the differences between the two variants.
  warnings.warn(

This is coming from transformer_prediction_interface.py which makes use of the checkpoints function.