raise StopIteration #20
Description
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??