We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 179a54a commit d526c2fCopy full SHA for d526c2f
README.md
@@ -36,6 +36,8 @@ For instance, run the following command to train a debiased encoder.
36
python main.py --tau_plus = 0.1
37
```
38
39
+#### *Due to the implementation of ```nn.DataParallel()```, training with at most 2 GPUs gives the best result.
40
+
41
## Linear evaluation
42
The model is evaluated by training a linear classifier after fixing the learned embedding.
43
0 commit comments