I think there are some errors in the mt_end_model.py

[WenxiongLiao]

When I use LSTMModule as the input_modules ,there will be a error: UnboundLocalError: local variable 'output_dim' referenced before assignment

so, I check the code of metal.multitask.mt_end_model.py
I found the local variable 'output_dim' really referenced before assignment at line 115.
I changed this line to output_dim=self.config["layer_out_dims"][0] , the input_modules can work.

But I am not sure if I am correct. I hope the author can check the code and it is best to give a tutorial of LSTMModule using as input_modules.

[bhancock8]

Thanks for bringing this to our attention! It looks like you were using the LSTM correctly. This should be fixed by PR #211.