最近遇到个问题,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":
版权声明:本文为weixin_40937909原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。