GPT2对接Flask运行时报dtype('O')转int64类型转换TypeError错误
问题修复方案
根因定位
代码generate_text方法内的seed变量赋值末尾多了多余逗号:
seed = None,
Python语法中变量赋值后加尾逗号会自动将变量转换为元组类型,此时seed实际值为(None,)(单元素元组),而非预期的None。np.random.seed、tf.set_random_seed均不支持元组作为随机种子入参,类型校验失败触发本次报错。
修复方法
删除seed赋值语句末尾的逗号即可:
seed = None
修改后重新启动Flask服务即可正常运行。
内容的提问来源于stack exchange,提问作者DK26
相关产品推荐
相关产品推荐

