search

amazon-science / unconditional-time-series-diffusion

Official PyTorch implementation of TSDiff models presented in the NeurIPS 2023 paper "Predict, Refine, Synthesize: Self-Guiding Diffusion Models for Probabilistic Time Series Forecasting"
Apache License 2.0
127 stars 25 forks source link

training_epoch_end error #2

Closed jimmylihui closed 10 months ago

jimmylihui commented 10 months ago

When I run the code, the following error appear: NotImplementedError: Support for training_epoch_end has been removed in v2.0.0. TSDiff implements this method. You can use the on_train_epoch_end hook instead. To access outputs, save them in-memory as instance attributes. You can find migration examples in https://github.com/Lightning-AI/lightning/pull/16520.

abdulfatir commented 10 months ago

@jimmylihui Did you install the package as in the instructions? Do you encounter this issue with version pinned in https://github.com/amazon-science/unconditional-time-series-diffusion/blob/main/pyproject.toml?