updated requirements and fixed a wrong reference, as pointed out in #187

jrzaurin · Nov 3, 2023 · 59ab075 · 59ab075
1 parent 68c3621
commit 59ab075
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pytorch_widedeep/self_supervised_training/contrastive_denoising_trainer.py b/pytorch_widedeep/self_supervised_training/contrastive_denoising_trainer.py
@@ -33,7 +33,7 @@ class ContrastiveDenoisingTrainer(BaseContrastiveDenoisingTrainer):
     r"""This class trains a Contrastive, Denoising Self Supervised 'routine' that
     is based on the one described in
     [SAINT: Improved Neural Networks for Tabular Data via Row Attention and
-    Contrastive Pre-Training](https://arxiv.org/abs/2203.05556), their Figure 1.
+    Contrastive Pre-Training](https://arxiv.org/abs/2106.01342), their Figure 1.
 
     Parameters
     ----------

diff --git a/requirements.txt b/requirements.txt
@@ -7,8 +7,8 @@ spacy
 opencv-contrib-python
 imutils
 tqdm
-torch
-torchvision
+torch >= 2.0.0
+torchvision >= 0.15.0
 einops
 wrapt
 torchmetrics