这是indexloc提供的服务,不要输入任何密码
Skip to content

EfficientNet models from TensorFlow.Keras not being reproducible on GPU #47174

@mhaghighat

Description

@mhaghighat

After downloading an EfficientNet model from tensorflow.keras.applications.efficientnet, and retraining it on our own data, I've noticed that the results are not reproducible. The results are reproducible for other architectures like VGG16, ResNet101, InceptionV3, and InceptionResNetV2, but not for any of the EfficientNetBx models.

Please note that I've set all the following seeds, and even have tensorflow-determinism:

random.seed(1)
np.random.seed(1)
tf.random.set_seed(1)
os.environ['TF_CUDNN_DETERMINISTIC'] = TRUE
os.environ['TF_DETERMINISTIC_OPS'] = TRUE

TensorFlow Version: tensorflow-gpu==2.3

Metadata

Metadata

Assignees

Labels

TF 2.3Issues related to TF 2.3TF 2.4for issues related to TF 2.4comp:gpuGPU related issuescomp:kerasKeras related issuestype:bugBug

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions