From 516ed93d203f6f076f92d3ef92e28f9c29dcd1c5 Mon Sep 17 00:00:00 2001
From: lzzyzlbb <287246233@qq.com>
Date: Mon, 17 Jan 2022 15:24:41 +0800
Subject: [PATCH] fix consistency for pix2pix (#552)

---
 test_tipc/test_train_inference_python.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index 9481f02..be18495 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -247,6 +247,7 @@ else
                 fi
                 # run train
                 eval "unset CUDA_VISIBLE_DEVICES"
+                export FLAGS_cudnn_deterministic=True
                 eval $cmd
                 echo $cmd
                 status_check $? "${cmd}" "${status_log}"
-- 
GitLab