Skip to content
Snippets Groups Projects
Commit 3c57551f authored by Stefan Reck's avatar Stefan Reck
Browse files

fix for graph datasets in train mode

parent e6529df4
Branches update_ci
Tags v0.16.3
No related merge requests found
Pipeline #22832 passed
......@@ -405,7 +405,7 @@ class IOHandler(object):
for n, file_no_set in enumerate(self.yield_files(which)):
# the number of samples in the n-th file of all inputs
file_sizes_full[n] = [h5_get_number_of_rows(
file, datasets=[self.cfg.key_y_values, self.cfg.key_x_values])
file, datasets=[self.cfg.key_y_values])
for file in file_no_set.values()]
if not file_sizes_full[n].count(file_sizes_full[n][0]) == \
len(file_sizes_full[n]):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment