diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_se_resnet.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_se_resnet.py index acef2f9d8da8e64a1801ba318109716f5506feb9..5f8e51c6532e885f6333907917c7206a72c2ac32 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_se_resnet.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_se_resnet.py @@ -16,7 +16,6 @@ import logging import math import time import unittest - import numpy as np import paddle @@ -34,9 +33,14 @@ EPOCH_NUM = 1 PRINT_STEP = 2 STEP_NUM = 10 -place = fluid.CPUPlace() -# TODO(liym27): Diff exists between dygraph and static graph on CUDA place. -# place = fluid.CUDAPlace(0) if fluid.is_compiled_with_cuda() else fluid.CPUPlace() +place = fluid.CUDAPlace(0) if fluid.is_compiled_with_cuda() \ + else fluid.CPUPlace() + +# Note: Set True to eliminate randomness. +# 1. For one operation, cuDNN has several algorithms, +# some algorithm results are non-deterministic, like convolution algorithms. +if fluid.is_compiled_with_cuda(): + fluid.set_flags({'FLAGS_cudnn_deterministic': True}) train_parameters = { "learning_strategy": {