dengyang17 / KEMI

10 stars 0 forks source link

Evaluation Metric For MI Dataset #4

Open lsyysl9711 opened 1 year ago

lsyysl9711 commented 1 year ago

The infer.py works smoothly for Esconv dataset while it is stuck in another MI dataset.

9dbc0efa78c2c5a1db0aedd72bd7a53

So in this line of code, we need to use one slot called "situation". But actually, there is no slot called "situation":

c345a3a55f8aaae7bb80137a0afda22

lsyysl9711 commented 1 year ago

CUDA_VISIBLE_DEVICES=3 python infer.py \ --config_name strat \ --inputter_name strat \ --data_name mi \ --knowledge_name sbert \ --add_nlg_eval \ --add_mi_analysis \ --seed 13 \ --load_checkpoint "/mnt/storage/lsy/gdown/blenderbot/DATA/strat.strat.mi.sbert/2023-09-22111813.3e-05.16.1gpu/epoch-2.bin" \ --fp16 false \ --max_input_length 256 \ --max_decoder_input_length 40 \ --max_length 40 \ --min_length 15 \ --infer_batch_size 2 \ --infer_input_file ./_reformat/ \ --temperature 0.7 \ --top_k 30 \ --top_p 0.3 \ --num_beams 1 \ --repetition_penalty 1 \ --no_repeat_ngram_size 3

Here is my inference script.

lsyysl9711 commented 1 year ago

I might find out the possible reason behind this: in MI dataset, we do not have a slot called "situation". So how should we fix this problem?