tkipf/gcn

How to make gcn run only using 1 thread and 4 thread?

AbhishekUoR opened this issue · 0 comments

Currently, GCN on CPU is using all threads of the machine, I use the following command "ps -o nlwp " to know how many threads it is using, the number comes out to be 51 on 48 hardware threads machine

I want to make it run only on 1 thread and 4 thread to see its performance, I tried inserting following TensorFlow code in train.py file
"""
session_conf = tf.ConfigProto(intra_op_parallelism_threads=1, inter_op_parallelism_threads=1)
sess = tf.Session(config=session_conf)
"""

But still, it uses 51 threads, is there any part of code in gcn directory wherein I can control the number of threads?