2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Moved general purpose functions up

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-11 16:23:32 +02:00
parent d53a0cb6b4
commit 0d527cf26d
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/twomartens/masterthesis/cli.py
View File

@ -368,6 +368,14 @@ def _pickle(filename: str, content: Any) -> None:
pickle.dump(content, 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]: def _ssd_evaluate_unbatch(glob_string: str) -> List[np.ndarray]:
import glob import glob
import pickle import pickle
@ -668,14 +676,6 @@ def _ssd_get_tensorboard_callback(args: argparse.Namespace, save_summaries_on_de
return tensorboard_callback 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, def _ssd_train_call(args: argparse.Namespace, train_function: callable,
train_generator: Generator, nr_batches_train: int, train_generator: Generator, nr_batches_train: int,
val_generator: Generator, nr_batches_val: int, val_generator: Generator, nr_batches_val: int,