【发布时间】:2021-11-11 12:17:01
【问题描述】:
我正在尝试使用来自https://github.com/daniel-kukiela/nmt-chatbot 的 nmt-chatbot,但是在使用自定义数据训练模型时出现错误,因为我在 google 上搜索这是因为在 Tensorflow v2 中删除了“contrib”模块,可以有人建议我替换那部分代码,
return tf.contrib.training.HParams(
AttributeError: module 'tensorflow' has no attribute 'contrib'
def create_hparams(flags):
"""Create training hparams."""
return tf.contrib.training.HParams(
# Data
src=flags.src,
tgt=flags.tgt,
train_prefix=flags.train_prefix,
dev_prefix=flags.dev_prefix,
test_prefix=flags.test_prefix,
vocab_prefix=flags.vocab_prefix,
embed_prefix=flags.embed_prefix,
out_dir=flags.out_dir,
【问题讨论】:
标签: python tensorflow artificial-intelligence tensorflow2.0 chatbot