我正在尝试建立一个VAE网络,我希望模型以不同的模式做不同的事情。我有三种模式:“train”,“same”和“different”,还有一个名为interpolation(mode)的函数,它根据模式做不同的事情。我的代码看起来像:
import tensorflow as tf
### some code here
mode = tf.placeholder(dtype = tf.string, name = "mode")
def interpolation(mode):
if mode == "train":
# do something
print("enter train mode")
elif mode == "same":
# do other things
print("enter same mode")
else:
# do other things
print("enter different mode")
# some other code here
sess.run(feed_dict = {mode: "train"})
sess.run(feed_dict = {mode: "same"})
sess.run(feed_dict = {mode: "different"})
但输出结果如下:
enter different mode
enter different mode
enter different mode
这意味着传入的模式不会改变条件。我做错了什么?如何通过字符串参数选择模式?你知道吗
第一种方法:您可以使用本机Tensorflowswitch-case选择不同的模式。例如,我假设你有三个案例,那么你可以做:
第二种方法:下面是另一种使用新AutoGraph API的方法:
相关问题 更多 >
编程相关推荐