Merge remote-tracking branch 'remotes/origin/master' into temp3 (aef96a82) · Commits · 钟慕尧 / deepchem

deepchem/models/tensorgraph/models/graph_models.py

+3 −1

Original line number	Diff line number	Diff line
		@@ -585,7 +585,9 @@ class GraphConvTensorGraph(TensorGraph):
		print('Starting epoch %i' % epoch)
		for ind, (X_b, y_b, w_b, ids_b) in enumerate(
		dataset.iterbatches(
		self.batch_size, pad_batches=True, deterministic=deterministic)):
		self.batch_size,
		pad_batches=pad_batches,
		deterministic=deterministic)):
		d = {}
		for index, label in enumerate(self.my_labels):
		if self.mode == 'classification':