Skip to content

Latest commit

 

History

History
29 lines (27 loc) · 1.1 KB

README.md

File metadata and controls

29 lines (27 loc) · 1.1 KB

Download

There are two branches in this REPO. The backup branch is deprecated and used as backup. You can only clone this main branch by:

git clone -b main --single-branch https://github.com/namespace-Pt/News-Recommendation.git

Instruction

  1. Download MIND dataset here
  2. Save MIND dataset in a directory, e.g. ~/Data
  3. pip install torch==1.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
  4. Add your email information so that you can get an email every time the model is evaluated: 
    cd src
mkdir data
echo "email = '[email protected]'" >> data/email.py
echo "password = 'your password'" >> data/email.py
  5. python -m main.twotower --data-root ~/Data   \
                        --batch-size 8       \
                        --world-size 2       \
                        --news-encoder bert
    • --world-size defines number of gpus in ddp training
    • --news-encoder defines news encoder
    • more parameters can be found in src/utils/manager.py