ant-research / Pyraformer

Apache License 2.0
264 stars 41 forks source link

what's the usage of predict_step in single_step_main.py ? #27

Open mw66 opened 1 year ago

mw66 commented 1 year ago

although, it's commented as # Test parameter

it's also used to load the TrainDataset: https://github.com/ant-research/Pyraformer/blob/master/single_step_main.py#L21

    train_set = eval(dataset+'TrainDataset')(data_dir, dataset, opt.predict_step, opt.inner_batch)

Is this an overlook, or it's by design. If by design, why we want to train with predict_step=24 by default in (single_step)_main?

Thanks.