Commit 751bfcd3 authored by ZHENQIN WU's avatar ZHENQIN WU
Browse files

fix travis failure

parent 4e63fa40

deepchem/models/tensorgraph/layers.py

+1 −1
Original line number Diff line number Diff line
@@ -1565,7 +1565,7 @@ class SigmoidCrossEntropy(Layer): 
  def __init__(self, in_layers=None, **kwargs):

    super(SigmoidCrossEntropy, self).__init__(in_layers, **kwargs)

    try:

      self._shape = self.in_layers[1].shape[:-1]

      self._shape = self.in_layers[1].shape

    except:

      pass

CRA Git | Maintained and supported by SUSTech CRA and CCSE