riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[Time Series] use mean scaler when scaling is a boolean True (#24237)

Browse Source 
* use mean scaler when scaling is boolean True

* remove debug
This commit is contained in:
Kashif Rasul 2023-06-13 18:46:05 +02:00 committed by GitHub
parent 695928e1e5
commit 4ed075280c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/autoformer/modeling_autoformer.py
View File

@ -1495,7 +1495,7 @@ class AutoformerModel(AutoformerPreTrainedModel):
def __init__(self, config: AutoformerConfig):
super().__init__(config)
if config.scaling == "mean" or config.scaling:
if config.scaling == "mean" or config.scaling is True:
self.scaler = AutoformerMeanScaler(dim=1, keepdim=True)
elif config.scaling == "std":
self.scaler = AutoformerStdScaler(dim=1, keepdim=True)

2
src/transformers/models/informer/modeling_informer.py
View File

@ -1504,7 +1504,7 @@ class InformerModel(InformerPreTrainedModel):
def __init__(self, config: InformerConfig):
super().__init__(config)
if config.scaling == "mean" or config.scaling:
if config.scaling == "mean" or config.scaling is True:
self.scaler = InformerMeanScaler(dim=1, keepdim=True)
elif config.scaling == "std":
self.scaler = InformerStdScaler(dim=1, keepdim=True)

2
src/transformers/models/time_series_transformer/modeling_time_series_transformer.py
View File

@ -1229,7 +1229,7 @@ class TimeSeriesTransformerModel(TimeSeriesTransformerPreTrainedModel):
def __init__(self, config: TimeSeriesTransformerConfig):
super().__init__(config)
if config.scaling == "mean" or config.scaling:
if config.scaling == "mean" or config.scaling is True:
self.scaler = TimeSeriesMeanScaler(dim=1, keepdim=True)
elif config.scaling == "std":
self.scaler = TimeSeriesStdScaler(dim=1, keepdim=True)