@ -191,8 +191,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
|
|||||||
|
|
||||||
# save predictions batch-wise to prevent memory problems
|
# save predictions batch-wise to prevent memory problems
|
||||||
if nr_digits is not None:
|
if nr_digits is not None:
|
||||||
counter = str(counter).zfill(nr_digits)
|
counter_str = str(counter).zfill(nr_digits)
|
||||||
filename = f"{output_file}-{counter}.npy"
|
filename = f"{output_file}-{counter_str}.npy"
|
||||||
else:
|
else:
|
||||||
filename = f"{output_file}-{counter:d}.npy"
|
filename = f"{output_file}-{counter:d}.npy"
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user