Skip to content

Commit f606324

Browse files
committed
fix: 🐛 a setup graph bug in the evaluation process
1 parent ab8d29d commit f606324

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

basicts/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from .launcher import launch_evaluation, launch_training
22
from .runners import BaseEpochRunner
33

4-
__version__ = '0.4.6.2'
4+
__version__ = '0.4.6.3'
55

66
__all__ = ['__version__', 'launch_training', 'launch_evaluation', 'BaseEpochRunner']

basicts/launcher.py

+4
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,10 @@ def evaluation_func(cfg: Dict,
4040
# initialize the logger for the runner
4141
runner.init_logger(logger_name='easytorch-evaluation', log_file_name='evaluation_log')
4242

43+
# setup the graph if needed
44+
if runner.need_setup_graph:
45+
runner.setup_graph(cfg=cfg, train=False)
46+
4347
try:
4448
# set batch size if provided
4549
if batch_size is not None:

0 commit comments

Comments
 (0)