From ed0afe832deb640162a5c822e65d2940c5231c42 Mon Sep 17 00:00:00 2001 From: Siarhei Siniak Date: Thu, 15 Jul 2021 08:31:01 +0300 Subject: [PATCH] [~] Refactor --- d1/jigsaw_toxic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/d1/jigsaw_toxic.py b/d1/jigsaw_toxic.py index 3bf07dc..f9dcb63 100644 --- a/d1/jigsaw_toxic.py +++ b/d1/jigsaw_toxic.py @@ -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 # %% [code] -%%time +#%%time with strategy.scope(): transformer_layer = ( transformers.TFDistilBertModel