correct typo

cold11 · Jan 2, 2020 · aae3a10 · aae3a10
1 parent ba38427
commit aae3a10
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/hanlp/callbacks/fine_csv_logger.py b/hanlp/callbacks/fine_csv_logger.py
@@ -9,7 +9,7 @@
 import tensorflow as tf
 
 
-class StreamTableFormater(object):
+class StreamTableFormatter(object):
 
     def __init__(self) -> None:
         super().__init__()
@@ -39,7 +39,7 @@ def __init__(self, filename, separator=',', append=False):
         self.filename = filename
         self.out: TextIOWrapper = None
         self.keys = []
-        self.formatter = StreamTableFormater()
+        self.formatter = StreamTableFormatter()
 
     def on_train_begin(self, logs=None):
         super().on_train_begin(logs)

diff --git a/hanlp/utils/tf_util.py b/hanlp/utils/tf_util.py
@@ -47,7 +47,7 @@ def set_gpu(idx=0):
     idx : int
         Which GPU to use
     """
-    gpus = get_visiable_gpus()
+    gpus = get_visible_gpus()
     if gpus:
         try:
             tf.config.experimental.set_visible_devices(gpus[idx], 'GPU')
@@ -59,13 +59,13 @@ def set_gpu(idx=0):
             raise e
 
 
-def get_visiable_gpus():
+def get_visible_gpus():
     gpus = tf.config.experimental.list_physical_devices('GPU')
     return gpus
 
 
 def set_gpu_memory_growth(growth=True):
-    gpus = get_visiable_gpus()
+    gpus = get_visible_gpus()
     if gpus:
         try:
             # Currently, memory growth needs to be the same across GPUs