Error in train.py

[albertovilla]

Hi, I think there is a bug in the train.py file, just when the main function starts the variable punctuation_enc is defined twice as you can see below. The second definition needs to be commented out in order to use train the model with LREC dataset.

   punctuation_enc = {
        'O': 0,
        'COMMA': 1,
        'PERIOD': 2,
        'QUESTION': 3
    }

    punctuation_enc = {
        'O': 0,
        'PERIOD': 1,
    }