Skip to content

Files

Latest commit

157b73e · Jun 4, 2023

History

History

recon_with_debias

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Jun 4, 2023
Jun 4, 2023
Jun 4, 2023
Jun 4, 2023

Environment

  • Python 3.7
  • Requirements
    • Install torch==1.9.0 according to your CUDA environment.
    • pip install -r requirements.txt
    • cd global_utils; pip install -e .

Data preparation

  • The GloVe embeddings can be downloaded from here.
  • The Google news embeddings can be downloaded from here.
  • The STS datasets can be found in ./examples/datasets/ in Huang's repository.

Experimental commands

python3 src/run_train.py --ref_vec_path <PATH> --result_dir <PATH> --lr 0.001 --adv_lr 0.001 --epoch 3 --batch_size 128  --save_selected_epoch 1_3 --adv_lambda 0.1 --save_vector --adv_freq_thresh 200000