Commit ec395116 authored by miaecle's avatar miaecle
Browse files

quick fix for gc

parent e2477458

deepchem/models/tensorgraph/models/graph_models.py

+1 −0
Original line number Diff line number Diff line
@@ -865,6 +865,7 @@ class GraphConvModel(TensorGraph): 
    return self.evaluate_generator(

        self.default_generator(dataset, predict=True),

        metrics,

        transformers=transformers,

        labels=self.my_labels,

        weights=[self.my_task_weights],

        per_task_metrics=per_task_metrics)

CRA Git | Maintained and supported by SUSTech CRA and CCSE