[~] Refactor
This commit is contained in:
parent
0d2aefa2b3
commit
ed0afe832d
@ -637,7 +637,7 @@ def build_model(transformer, max_len=512):
|
|||||||
# If you want to use any another model just replace the model name in transformers._____ and use accordingly
|
# If you want to use any another model just replace the model name in transformers._____ and use accordingly
|
||||||
|
|
||||||
# %% [code]
|
# %% [code]
|
||||||
%%time
|
#%%time
|
||||||
with strategy.scope():
|
with strategy.scope():
|
||||||
transformer_layer = (
|
transformer_layer = (
|
||||||
transformers.TFDistilBertModel
|
transformers.TFDistilBertModel
|
||||||
|
Loading…
Reference in New Issue
Block a user