c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Corrected logger.error info

Browse Source 
Signed-off-by: Kevin Trebing <Kevin.Trebing@gmx.net>
This commit is contained in:
Kevin Trebing authored and Lysandre Debut committed 2019-08-09 19:36:44 -04:00
1 parent 3566d27919
commit 7060766490
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

pytorch_transformers/modeling_bert.py
+1 -1
View File
@@ -74,7 +74,7 @@ def load_tf_weights_in_bert(model, config, tf_checkpoint_path):
import numpy as np
import tensorflow as tf
except ImportError:
logger.error("Loading a TensorFlow models in PyTorch, requires TensorFlow to be installed. Please see "
logger.error("Loading a TensorFlow model in PyTorch, requires TensorFlow to be installed. Please see "
"https://www.tensorflow.org/install/ for installation instructions.")
raise
tf_path = os.path.abspath(tf_checkpoint_path)
pytorch_transformers/modeling_gpt2.py
+1 -1
View File
@@ -50,7 +50,7 @@ def load_tf_weights_in_gpt2(model, config, gpt2_checkpoint_path):
import numpy as np
import tensorflow as tf
except ImportError:
logger.error("Loading a TensorFlow models in PyTorch, requires TensorFlow to be installed. Please see "
logger.error("Loading a TensorFlow model in PyTorch, requires TensorFlow to be installed. Please see "
"https://www.tensorflow.org/install/ for installation instructions.")
raise
tf_path = os.path.abspath(gpt2_checkpoint_path)