Commit 3d5977ef authored by panicPaul's avatar panicPaul
Browse files

merge commit

parent 9a11259d
......@@ -29,7 +29,6 @@ class ${tc.fileNameWithoutEnding}:
train_data[input_name] = train_h5[input_name]
train_dataset = train_h5[input_name]
train_dataset_shape = train_data[input_name].shape
# slice_size limits the memory consumption, by only loading slices of size <500MB into memory
slice_size = min(train_dataset_shape[0] - 1, int(500e6 / (train_h5[input_name][0].size * \
train_h5[input_name][0].itemsize)))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment