我尝试转换Seq2Seq model我在本地运行以使用Estimator和Experiment使用Tensorflow分配功能。基本功能和目标设置如下:
for every input and response (translation or prompt and response):
raw input -> tokenized input -> tokenized response -> raw response
注意:
几个问题:
train_steps
和eval_steps
与培训和评估的迭代有关。是否有其他选项可以为这些步骤设置批量大小,还是实验会在内部/自动计算批量大小?train_input_fn
和eval_input_fn
可以是任何返回特征字典和目标张量的input_fn。在上面的例子中,我真的只需要一个特征张量和一个目标张量,因为我创建一个自定义Estimator,只要我的Estimator model_fn
期望那些形状和罐头,它就可以是任何形状适当地从他们那里回来。这是对的吗?