Commit 54b178c5 authored by peastman's avatar peastman
Browse files

Fixed infinite loop

parent 4f4d72fb

deepchem/data/datasets.py

+2 −1
Original line number Diff line number Diff line
@@ -748,6 +748,7 @@ class DiskDataset(Dataset): 
        num_local_batches = math.ceil(n_shard_samples / shard_batch_size)



        if n_shard_samples == 0:

          cur_shard += 1

          continue

        if not deterministic:

          sample_perm = np.random.permutation(n_shard_samples)

CRA Git | Maintained and supported by SUSTech CRA and CCSE