Why can't I use MPC commands?

[chenci107]

When Ｉuse the command：
python3 train_variational_koopman.py --seq_length 16 --n_trials 20 --n_subseq 220 --kl_weight 0.1 --extractor_size 64 64 --inference_size 64 64 --prior_size 64 32 --ilqr True
There will be an error：ValueError: Cannot compute Hessian because element 0 of xs does not have rank one.. Tensor concat_23:0 must have rank 1. Received rank 2, shape (16, 5)