From 1ee7798fb9089ec6328cfcf04c6dd3606e8b4ecf Mon Sep 17 00:00:00 2001 From: iperov Date: Fri, 11 Dec 2020 17:56:39 +0400 Subject: [PATCH] . --- models/Model_XSeg/Model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Model_XSeg/Model.py b/models/Model_XSeg/Model.py index 5b26ba5..fe444d2 100644 --- a/models/Model_XSeg/Model.py +++ b/models/Model_XSeg/Model.py @@ -101,7 +101,7 @@ class XSegModel(ModelBase): # Average losses and gradients, and create optimizer update ops - with tf.device(f'/CPU:0'): + with tf.device(f'/CPU:0'): # Temporary fix. Unknown bug with training freeze starts from 2.4.0, but 2.3.1 was ok #with tf.device (models_opt_device): pred = tf.concat(gpu_pred_list, 0) loss = tf.concat(gpu_losses, 0)