search

MahmoudWahdan / dialog-nlu

Tensorflow and Keras implementation of the state of the art researches in Dialog System NLU
Apache License 2.0
98 stars 40 forks source link

Can not import BertCrfNLU #38

Closed zeinabfarhoudi closed 3 years ago

zeinabfarhoudi commented 3 years ago

Hi, after I import BertCrfNLU, I can't train with this module. would you please fix the problem

MahmoudWahdan commented 3 years ago

Hi @farhodi Bug fixed as in 1421d28ecdc3ccb1d1f08ef736f37ea1592abc69 Please, pull latest and let me know if it works with you.

zeinabfarhoudi commented 3 years ago

Hi @MahmoudWahdan , Thanks for your response. I get the latest one, but when I run BertCrfNLU to train the model, I face a memory leak error. I run the code in colab with TPU running mode. When I run to train the model, after displaying the summary of the model and before the first epoch training model begins, my session drop because of lack of memory!

MahmoudWahdan commented 3 years ago

Hi @farhodi Could you please put the code and error here? How much data do you have?

MahmoudWahdan commented 3 years ago

Hi @farhodi Did you get the chance to check this notebook https://github.com/MahmoudWahdan/dialog-nlu/blob/master/notebooks/using_bert_crf_nlu.ipynb