Change loss function for multi-category because CrossEntropyLoss only accept outputs of (batch, x) and labels with shape of (batch)

First commit

Found a critical bug in LSTM inference when distillation using BERT Tokenizer but inference uses custom LSTM tokenizer

Update LSTM gradient clipping

test

Save vocab size with model_state_dict

Edit some saving and loading details

Add LSTM fine tuning