diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 4cc1f44..bdc9951 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -366,6 +366,14 @@ def _pickle(filename: str, content: Any) -> None: with open(filename, "wb") as file: pickle.dump(content, file) + + +def _get_nr_batches(data_length: int, batch_size: int) -> int: + return int(math.floor(data_length / batch_size)) + + +def _get_nr_digits(data_length: int, batch_size: int) -> int: + return math.ceil(math.log10(math.ceil(data_length / batch_size))) def _ssd_evaluate_unbatch(glob_string: str) -> List[np.ndarray]: @@ -668,14 +676,6 @@ def _ssd_get_tensorboard_callback(args: argparse.Namespace, save_summaries_on_de return tensorboard_callback -def _get_nr_batches(data_length: int, batch_size: int) -> int: - return int(math.floor(data_length / batch_size)) - - -def _get_nr_digits(data_length: int, batch_size: int) -> int: - return math.ceil(math.log10(math.ceil(data_length / batch_size))) - - def _ssd_train_call(args: argparse.Namespace, train_function: callable, train_generator: Generator, nr_batches_train: int, val_generator: Generator, nr_batches_val: int,