Refactor test_prepare_checkpoints (openvinotoolkit#827)

* Required checkpoint path in `prepare_checkpoints.py` * Refactored test to load real checkpoints from sanity test_model_train to properly test `prepare_checkpoints.py`, which required checkpoint path. (It was impossible to make new small model to pass it into prepare checkpoint because prepare checkpoint can build only yolo, retinanet and maskrcnn models.) Local check shows new version of the test small time consuming (~30 sec ~= not refactored test running time).
kshpv · Jul 7, 2021 · 485140c · 485140c
1 parent df7ba60
commit 485140c
Show file tree

Hide file tree

Showing 7 changed files with 14 additions and 14 deletions.
diff --git a/examples/tensorflow/common/argparser.py b/examples/tensorflow/common/argparser.py
diff --git a/examples/tensorflow/common/prepare_checkpoint.py b/examples/tensorflow/common/prepare_checkpoint.py
diff --git a/tests/tensorflow/data/configs/resnet50_cifar10_magnitude_sparsity_int8.json b/tests/tensorflow/data/configs/resnet50_cifar10_magnitude_sparsity_int8.json
diff --git a/tests/tensorflow/data/configs/retinanet_coco2017_magnitude_sparsity_int8.json b/tests/tensorflow/data/configs/retinanet_coco2017_magnitude_sparsity_int8.json
diff --git a/tests/tensorflow/data/configs/sequential_model_cifar10_magnitude_sparsity_int8.json b/tests/tensorflow/data/configs/sequential_model_cifar10_magnitude_sparsity_int8.json
diff --git a/tests/tensorflow/data/configs/sequential_model_no_input_cifar10_magnitude_sparsity_int8.json b/tests/tensorflow/data/configs/sequential_model_no_input_cifar10_magnitude_sparsity_int8.json
diff --git a/tests/tensorflow/test_sanity_sample.py b/tests/tensorflow/test_sanity_sample.py