google-deepmind / google-deepmind/graph_nets

Performance issue

Open
#146 9 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
5.4k
Forks
777
PR merge metrics
No merged PRs in 30d

Description

I am using the graph module to solve an ETA estimation problem. Could anybody tell me how to make the training faster by using all the available cores on my computer? I tried many things in TensorFlow 2.5, such as setting the number of thread by tf.config.threading.set_inter_op_parallelism_threads and tf.config.threading.set_intra_op_parallelism_threads, but nothing works. Training the MLP networks took a very long time.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.