Moved general purpose functions up
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
parent
d53a0cb6b4
commit
0d527cf26d
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue