Commit 44a932cc authored by peastman's avatar peastman
Browse files

Bug fix

parent b9ad5c8a

deepchem/models/keras_model.py

+1 −0
Original line number Diff line number Diff line
@@ -342,6 +342,7 @@ class KerasModel(Model): 
          grads = tape.gradient(loss, self.model.trainable_variables)

          self._tf_optimizer.apply_gradients(

              zip(grads, self.model.trainable_variables))

          tf.assign_add(self._global_step, 1)

          current_step = self._global_step.numpy()

      else:

CRA Git | Maintained and supported by SUSTech CRA and CCSE