strongio / keras-bert

A simple technique to integrate BERT from tf hub to keras
258 stars 108 forks source link

fix issue n_layer_fine_tune #6

Closed kkkyan closed 5 years ago

kkkyan commented 5 years ago

Fixes #5 use some tricks to find the correct layer number and add the layer variables to trainable_variable