-
Notifications
You must be signed in to change notification settings - Fork 173
Open
Description
I need to run multiple TFFMRegressor objects in joblib Parallel. To do so, I passed the following parameter:
session_config=tf.ConfigProto(intra_op_parallelism_threads=1,
inter_op_parallelism_threads=1,
allow_soft_placement=True,
device_count = {'CPU': 1, 'GPU': 0})
However, my cores do not seem to run whenever I use n_jobs=2 or higher in Parallel; my python notebook cell just hangs, never completes and my processors are not used. At n_jobs=1, everything is running fine. What am I missing? Would I better use polylearn instead for this kind of task?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels