- 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 .
- Install
- 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.
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