text: Title: TiDE Model Stops At A Specific Epoch
  Body: A more general question, I am trying to run a historical backtest using TiDE model for my use case:
  
  ```
  from darts.models import TiDEModel
  
  tide_model = TiDEModel(
      input_chunk_length=8,
      output_chunk_length=3,
      n_epochs=20
  )
  
  tide_model .fit(
      series =...,
      past_covariates= ...
      future_covariates= ...
  )
      
  tide_hf_results = model_estimator.historical_forecasts(
  ...
  ) 
  ```
  For some reason, the model always stalls at a specific point (77% of Epoch 5). I can see that the kernel is still running under the hood but the progress bar will no longer continue moving. I have tried increasing the memory and CPU by 3x but still, the model would stall at exactly the same point. Not sure if anyone have met this issue before and have any suggested solutions. 
  
  No error messages are returned at all so I am not sure how to debug the issue.
  
  ![image](https://github.com/user-attachments/assets/60f961ca-f65f-44ab-a9e9-fffb8d0831d8)
  