Commit c545d3ab authored by peastman's avatar peastman
Browse files

Fixed test failure

parent 55d1ff20

deepchem/models/tensorgraph/progressive_multitask.py

+1 −1
Original line number Diff line number Diff line
@@ -247,7 +247,7 @@ class ProgressiveMultitaskRegressor(KerasModel): 
        dataset, epochs=nb_epoch, deterministic=deterministic)

    variables = []

    for layer in self._task_layers[task]:

      variables.append(layer.trainable_variables)

      variables += layer.trainable_variables

    loss = TaskLoss(self.model, self.create_loss(), task)

    self.fit_generator(

        generator,

CRA Git | Maintained and supported by SUSTech CRA and CCSE