Remove disable_validation param

minimaxir · Jun 28, 2020 · e0848e1 · e0848e1
1 parent fc9d738
commit e0848e1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/aitextgen/aitextgen.py b/aitextgen/aitextgen.py
@@ -528,7 +528,6 @@ def train(
             gradient_clip_val=max_grad_norm if not fp16 else 0,
             checkpoint_callback=False,
             logger=loggers if loggers else False,
-            disable_validation=True,
             weights_summary=None,
             callbacks=[
                 ATGProgressBar(

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="aitextgen",
     packages=["aitextgen"],  # this must be the same as the name above
-    version="0.2.0",
+    version="0.2.1",
     description="A robust Python tool for text-based AI training and generation using GPT-2.",
     long_description=open("README.md", "r", encoding="utf-8").read(),
     long_description_content_type="text/markdown",