Commit 8fba0cd5 authored by Anna Warno's avatar Anna Warno
Browse files

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

parent 64f4675e
Pipeline #17013 passed with stage
in 1 minute and 11 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