jit compiled error with DeepAR prediction net

zalandoresearch / pytorch-ts

PyTorch based Probabilistic Time Series forecasting framework based on GluonTS backend

MIT License

1.21k stars 190 forks source link

Closed rubick1896 closed 3 years ago

rubick1896 commented 3 years ago

I want to compile a trained DeepAR prediction net with

net = predictor.prediction_net scripted_module = torch.jit.script(net)

and got the following error: Compiled functions can't take variable number of arguments or use keyword-only arguments with defaults

kashif commented 3 years ago

@rubick1896 at the moment jit is not supported sorry!