pso2keras 버전을 1.0.5에서 1.0.5.1로 업데이트하고, TensorFlow GPU 메모리 성장을 설정하고, TF_CPP_MIN_LOG_LEVEL을 2로 설정했습니다.

pso/__init__.py

@@ -1,10 +1,21 @@
+import tensorflow as tf
+import os
 from .optimizer import Optimizer as optimizer
 from .particle import Particle as particle
 
-__version__ = "1.0.5"
+__version__ = "1.0.5.1"
 
 print("pso2keras version : " + __version__)
 
+gpus = tf.config.experimental.list_physical_devices("GPU")
+if gpus:
+    try:
+        tf.config.experimental.set_memory_growth(gpus[0], True)
+    except RuntimeError as r:
+        print(r)
+
+os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
+
 __all__ = [
     "optimizer",
     "particle",