fix: 🐛 a bug in utsf runner

Author: zezhishao

basicts/__init__.py

@@ -1,6 +1,6 @@
 from .launcher import launch_evaluation, launch_inference, launch_training
 from .runners import BaseEpochRunner
 
-__version__ = '0.5.5'
+__version__ = '0.5.6'
 
 __all__ = ['__version__', 'launch_training', 'launch_evaluation', 'BaseEpochRunner', 'launch_inference']

basicts/runners/base_utsf_runner.py

@@ -57,7 +57,7 @@ def __init__(self, cfg: Dict):
         self.grad_accumulation_steps = cfg.get('TRAIN.GRAD_ACCUMULATION_STEPS',1)
 
         # inference settings
-        self.generation_params = cfg['INFERENCE'].get('GENERATION_PARAMS', {})
+        self.generation_params = cfg.get('INFERENCE', {}).get('GENERATION_PARAMS', {})
         self.forward_features = [0] # do not use time features
         self.target_features = [0] # do not use time features
 
@@ -358,8 +358,9 @@ def val_iters(self, iter_index: int, data: Union[torch.Tensor, Tuple]):
         data = self.preprocessing(data)
         # TODO: consider using amp for validation
         # with self.ctx:
-        forward_return = self.forward(data=data, iter_num=iter_index, train=False)
-        forward_return = self.postprocessing(forward_return)
+        with self.amp_context:
+            forward_return = self.forward(data=data, iter_num=iter_index, train=False)
+            forward_return = self.postprocessing(forward_return)
         loss = self.metric_forward(self.loss, forward_return)
         self.update_iteration_meter('val/loss', loss)