Skip to content

Chronos: support train_dataloader in Forecaster.tune #4862

Open
@shane-huang

Description

@shane-huang

Problem:

  • BasePytorchForecaster.tune (src code) currently only support numpy array tuple as input, so that batch size can be tuned.

Solution:

We could also support train_dataloaders as input. Just in this case, batch size cannot be tuned option.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions