如何改变Tensorflow 2中的策略

2024-10-03 02:32:46 发布

您现在位置:Python中文网/ 问答频道 /正文

我想更改用于Tensorflow 2对象检测API培训和评估的GPU

我试图改变评估和训练的策略。对于评估,我只需要使用1个GPU

strategy = tf.distribute.MirroredStrategy(["GPU:0"])

但是火车要两个

我得到这个错误:

"RuntimeError: Mixing different tf.distribute.Strategy objects: <tensorflow.python.distribute.mirrored_strategy.MirroredStrategy object at

有什么建议吗


Tags: 对象apigputftensorflow错误策略distribute