hyeongjun-jang / crs_212ho

KT Ho

LATTE

Recommendation Task

python tester.py \
  --batch_size=32 \
  --name=xxx \  # Log Name
  --kg_emb_dim=128 \
  --meta=meta-word \
  --n_meta=4 \
  --loss_lambda=0.8 \
  --lr_ft=1e-3 \
  --dropout_pt=0.25 \
  --dropout_ft=0.2 \
  --task=rec \
  --dataset_path=data/redial \  # data/inspired for INSPIRED
  --device_id=0 #GPU device id

Conversational Task

python main.py \
   --conv_batch_size=8 \
   --name=xxx \ # Log Name
   --gpt_name=gpt2-large \
   --conv_lr_ft=1e-3 \
   --conv_loss_lambda=2.0 \
   --task=conv \
   --max_gen_len=100 \
   --max_title_len=100 \
   --context_max_length=32 \
   --n_template_sample=2 \
   --dataset_path=data/redial \  # data/inspired for INSPIRED
   --device_id=0