How to run self supervised pretraining using UniTS on custom data?

[nikhilajoshy]

No description provided.

[gasvn]

https://github.com/mims-harvard/UniTS/blob/main/Tutorial.md
You can refer to this. The major difference is that you need to define your dataset in the config in such way: https://github.com/mims-harvard/UniTS/blob/main/data_provider/multi_task_pretrain.yaml

[wbzhang233]

How to make inference for forecasting task with pretrained checkpoints?

大佬，请问怎么用UniTS模型来做预测呢？我看只有脚本里面把is_training改成0就可以实现预测，但是我们需要的不是训练后，输入新的序列，直接预测接下来的序列吗？只有那个脚本来做test感觉不太好用啊。