我是tensorflow的新手,最近我正在处理一个项目,该项目有以下代码
def build_fv_graph(self) -> tf.Tensor:
with tf.variable_scope("modifier_attr"):
t_mdl_name = tf.constant(self.model_name, name="mdl_name", dtype=tf.string)
字符串
正如你所看到的,这里的代码使用tf.constant
将一个字符串硬编码到模型中,这将方便用户覆盖它,例如,通过环境变量。我的问题是这是可能的,或者对于这种用例有更好的解决方案吗?
1条答案
按热度按时间0sgqnhkj1#
我仍然不确定这是否对你有用。但是让我们试试。用这样的东西加载你的模型:
字符串
当创建模型时,使用类似这样的东西:
型
当你想改变变量时,使用类似这样的方法:
型
这就是我知道你提供的细节所能想到的。也许有些语法是不正确的,但想法就在这里。