Merge branch 'GestaltCogTeam:master' into contribution

superarthurlx · Jan 24, 2025 · b6da3f8 · b6da3f8
2 parents 9c0cf6d + f606324
commit b6da3f8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/basicts/__init__.py b/basicts/__init__.py
@@ -1,6 +1,6 @@
 from .launcher import launch_evaluation, launch_training
 from .runners import BaseEpochRunner
 
-__version__ = '0.4.6.2'
+__version__ = '0.4.6.3'
 
 __all__ = ['__version__', 'launch_training', 'launch_evaluation', 'BaseEpochRunner']
diff --git a/basicts/launcher.py b/basicts/launcher.py
@@ -40,6 +40,10 @@ def evaluation_func(cfg: Dict,
     # initialize the logger for the runner
     runner.init_logger(logger_name='easytorch-evaluation', log_file_name='evaluation_log')
 
+    # setup the graph if needed
+    if runner.need_setup_graph:
+        runner.setup_graph(cfg=cfg, train=False)
+
     try:
         # set batch size if provided
         if batch_size is not None: