Closed RoelofKuipers closed 4 months ago
Hey @RoelofKuipers, thanks for using neuralforecast. That argument refers to the feature added in #901, so it's not present in 1.6.4. If you want to try the new models I suggest installing from github instead, so that you get a compatible version.
@RoelofKuipers I have just made a quick test using the dev env for TSMixerx
and it works fine with the optimizer
argument. Hope this helps
Thank you :)
What happened + What you expected to happen
I tried to run to use the TSMixer.py class, but got an error :
TypeError: Trainer.__init__() got an unexpected keyword argument 'optimizer'
I think it is because because of the pytorch-lightning does not have anoptimizer
argument in theTrainer
constructor anymore. This applies to these two lines in tsmixerx.pyVersions / Dependencies
pytorch-lightning=2.0.0 torch==2.1.0 neuralforecast==1.6.4
Reproduction script
Issue Severity
Medium: It is a significant difficulty but I can work around it.