Skip to content

Commit d526c2f

Browse files
authored
Update README.md
1 parent 179a54a commit d526c2f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

README.md

+2
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@ For instance, run the following command to train a debiased encoder.
3636
python main.py --tau_plus = 0.1
3737
```
3838

39+
#### *Due to the implementation of ```nn.DataParallel()```, training with at most 2 GPUs gives the best result.
40+
3941
## Linear evaluation
4042
The model is evaluated by training a linear classifier after fixing the learned embedding.
4143

0 commit comments

Comments
 (0)