Commit bb842aa5 authored by Marta Różańska's avatar Marta Różańska
Browse files

Merge branch 'tft_nbeats' into 'morphemic-rc1.5'

deafult number of workers for tft and nbetas dataloaders changed (6 -> default low number)

See merge request !197
parents c4b3a507 8fba0cd5
Pipeline #17015 canceled with stages
in 16 seconds
......@@ -87,10 +87,8 @@ def train(target_column, prediction_length, yaml_file="model.yaml", publish_rate
bs = params["training"]["bs"]
train_dataloader = training.to_dataloader(
train=True, batch_size=bs, num_workers=6, shuffle=True
)
val_dataloader = validation.to_dataloader(train=False, batch_size=bs, num_workers=6)
train_dataloader = training.to_dataloader(train=True, batch_size=bs, shuffle=True)
val_dataloader = validation.to_dataloader(train=False, batch_size=bs)
early_stop_callback = EarlyStopping(
monitor="train_loss", min_delta=1e-5, patience=8, verbose=False, mode="min"
......
......@@ -87,10 +87,8 @@ def train(target_column, prediction_length, yaml_file="model.yaml", publish_rate
)
bs = params["training"]["bs"]
train_dataloader = training.to_dataloader(
train=True, batch_size=bs, num_workers=6, shuffle=True
)
val_dataloader = validation.to_dataloader(train=False, batch_size=bs, num_workers=6)
train_dataloader = training.to_dataloader(train=True, batch_size=bs, shuffle=True)
val_dataloader = validation.to_dataloader(train=False, batch_size=bs)
early_stop_callback = EarlyStopping(
monitor="val_loss", min_delta=1e-5, patience=8, verbose=False, mode="min"
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment