This program (convert_pkl_to_csv.py) converts attention weights in pickle format to CSV format. 
The attention weight in pickle format has five dimensions (sample, layer, head, sequence, sequence).
In addition to the input file path and output directory, Users need to specify the sample number, layer number, and head number of the output attention weight.

execution)
python convert_pkl_to_csv.py

option)
-i input file path
-o output directory
-s sample number (1~the number of samples)
-l layer number (1~3)
-n head number (1~4)

ex)
python convert_pkl_to_csv.py -i ./attention_weight.pkl -o ./ -s 1 -l 1 -n 1

