wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-22 13:00:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

bug fixes in NBEATSEstimator (#74)

Browse Source 
* spell correction in n_beats_ensemble

* bug fix - NBEATSEstimator intialization
This commit is contained in:
Shashank Deshpande
2021-09-02 09:53:09 +02:00
committed by GitHub
parent 16f0f9ef8d
commit 0b0dfe12ca
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/n_beats/n_beats_ensemble.py
+1 -1
View File
@@ -170,7 +170,7 @@ class NBEATSEnsembleEstimator(PyTorchEstimator):
Arguments passed down to the individual estimators.
"""
@validted()
@validated()
def __init__(
self,
freq: str,
pts/model/n_beats/n_beats_estimator.py
+1 -1
View File
@@ -30,7 +30,7 @@ from .n_beats_network import (
)
class NBEATSEstimator:
class NBEATSEstimator(PyTorchEstimator):
@validated()
def __init__(
self,