Commit 70898a69 authored by ZHENQIN WU's avatar ZHENQIN WU
Browse files

little changes

parent 215a45f1

deepchem/models/tensorflow_models/model_ops.py

+1 −1
Original line number Diff line number Diff line
@@ -181,7 +181,7 @@ def weight_decay(penalty_type, penalty): 


  with tf.name_scope('weight_decay'):

    if penalty_type == 'l1':

      cost = tf.add_n([tf.reduce_sum(tf.Abs(v)) for v in variables])

      cost = tf.add_n([tf.reduce_sum(tf.abs(v)) for v in variables])

    elif penalty_type == 'l2':

      cost = tf.add_n([tf.nn.l2_loss(v) for v in variables])

    else:

CRA Git | Maintained and supported by SUSTech CRA and CCSE