Loading deepchem/trans/transformers.py +1 −0 Original line number Diff line number Diff line Loading @@ -677,6 +677,7 @@ class IRVTransformer(): values = [] top_labels = [] with g_temp.as_default(): with tf.device('/cpu:0'): labels_tf = tf.constant(y) similarity_placeholder = tf.placeholder( dtype=tf.float64, shape=(None, reference_len)) Loading Loading
deepchem/trans/transformers.py +1 −0 Original line number Diff line number Diff line Loading @@ -677,6 +677,7 @@ class IRVTransformer(): values = [] top_labels = [] with g_temp.as_default(): with tf.device('/cpu:0'): labels_tf = tf.constant(y) similarity_placeholder = tf.placeholder( dtype=tf.float64, shape=(None, reference_len)) Loading
