TypeError: while_loop() got an unexpected keyword argument 'maximum_iterations'

绝地灬酷狼 2021-12-19 23:35 311阅读 0赞

转载自:https://blog.csdn.net/zyh2004883/article/details/84337872

错误:

TypeError: while_loop() got an unexpected keyword argument ‘maximum_iterations’

参照https://blog.csdn.net/qq_27009517/article/details/83275846

重新安装keras

因为我的CUDA版本为CUDA8.0,所以tensorflow版本有限制,只能用1.4.0以下的版本

所以tensorflow-gpu=1.4.0

之前keras版本太高了,所以才会报以上错误

更新keras版本后,错误解决

pip install keras==2.1.2

  1. # -*- coding: utf-8 -*-
  2. # @Time : 2019/7/1 8:17
  3. # @Author : ZXL
  4. # @Site :
  5. # @File : many_input_API.py
  6. # @Software: PyCharm
  7. from keras.models import Model
  8. from keras import layers
  9. from keras import Input
  10. text_vocabulary_size = 10000
  11. question_vocabulary_size = 10000
  12. answer_vocabulary_size = 500
  13. text_input = Input(shape=(None,), dtype='int32', name='text')
  14. embedded_text = layers.Embedding(64, text_vocabulary_size)(text_input)
  15. encoded_text = layers.LSTM(32)(embedded_text)
  16. question_input = Input(shape=(None,),dtype='int32',name='question')
  17. embedded_question = layers.Embedding(32, question_vocabulary_size)(question_input)
  18. encoded_question = layers.LSTM(16)(embedded_question)
  19. concatenated = layers.concatenate([encoded_text, encoded_question],axis=-1)#两个输入处理结果联系起来
  20. answer = layers.Dense(answer_vocabulary_size,activation='softmax')(concatenated)
  21. model = Model([text_input, question_input], answer)#模型声明
  22. model.compile(optimizer='rmsprop',loss='categorical_crossentropy',metrics=['acc'])#compile阶段
  23. model.summary()

在这里插入图片描述

发表评论

表情:
评论列表 (有 0 条评论,311人围观)

还没有评论,来说两句吧...

相关阅读