raise StopIteration

[fmorenovr]

I ran "python main.py" and I got:

labelprobe image index 63296, items per sec 99.2679, 126.2880

Traceback (most recent call last):
File "/home/jenazads/github/NetDissect-Lite/loader/data_loader.py", line 518, in batches
raise StopIteration
StopIteration

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "main.py", line 18, in
tally_result = fo.tally(features[layer_id],thresholds,savepath="tally.csv")
File "/home/jenazads/github/NetDissect-Lite/feature_operation.py", line 216, in tally
FeatureOperator.tally_job((features, self.data, threshold, tally_labels, tally_units, tally_units_cat, tally_both, 0, self.data.size()))
File "/home/jenazads/github/NetDissect-Lite/feature_operation.py", line 142, in tally_job
for batch in pd.batches():
RuntimeError: generator raised StopIteration

I identify that it happens in the feature_operation.py when batch is None.
What should I do? or do you know what is happening??

[aaossa]

I think this is a duplicated issue: #18 and #14

Maybe this answer (#14 (comment)) works for you

[fmorenovr]

I think this is a duplicated issue: #18 and #14

Maybe this answer (#14 (comment)) works for you

I already solved it. But Thanks!!

[2021202420]

Hey,I also encountered this problem, how did solve the problem? can you please share?