LSTM的训练和测试长度不一样

最近遇到个问题，LSTM的训练的样本的时间长度是一致的，但是测试的时候用到生成器则希望其长度可以不定（根据生成的Latent space的长度不同而不同），搜了下好像没有特别好的方法（有的是训练的时候就改成timestep是None）但是我现在已经训练好了。

后来试了下去改model就可以了，

{
    "class_name": "Model",
    "config": {
        "name": "model_1",
        "layers": [{
            "name": "input_1",
            "class_name": "InputLayer",
            "config": {
                "batch_input_shape": [null, 100, 5],
                "dtype": "float32",
                "sparse": false,
                "name": "input_1"
            },
            "inbound_nodes": []
        }, {
            "name": "generator",
            "class_name": "Sequential",
            "config": [{
                "class_name": "LSTM",
                "config": {
                    "name":