From 485140cb50c80cf8708335c3a9ac1adc9ac89359 Mon Sep 17 00:00:00 2001 From: Daniil Lyakhov Date: Wed, 7 Jul 2021 18:37:51 +0300 Subject: [PATCH] Refactor test_prepare_checkpoints (#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). --- examples/tensorflow/common/argparser.py | 4 ++-- examples/tensorflow/common/prepare_checkpoint.py | 4 ++-- .../configs/resnet50_cifar10_magnitude_sparsity_int8.json | 4 ++-- .../configs/retinanet_coco2017_magnitude_sparsity_int8.json | 4 ++-- .../sequential_model_cifar10_magnitude_sparsity_int8.json | 4 ++-- ...ential_model_no_input_cifar10_magnitude_sparsity_int8.json | 4 ++-- tests/tensorflow/test_sanity_sample.py | 4 ++-- 7 files changed, 14 insertions(+), 14 deletions(-) diff --git a/examples/tensorflow/common/argparser.py b/examples/tensorflow/common/argparser.py index 31777fbb35c..b7faab2b54e 100644 --- a/examples/tensorflow/common/argparser.py +++ b/examples/tensorflow/common/argparser.py @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:acfcbe8d94052a0fa6945d639defb61e57afd50c3641ce11f66de6815a53309a -size 7112 +oid sha256:99bfce895a450ef147576792d8e9b5d366fcf408458256cf702083be0c72e6a8 +size 7235 diff --git a/examples/tensorflow/common/prepare_checkpoint.py b/examples/tensorflow/common/prepare_checkpoint.py index 2e55840a27f..65442a5f9b2 100644 --- a/examples/tensorflow/common/prepare_checkpoint.py +++ b/examples/tensorflow/common/prepare_checkpoint.py @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:63ee29e7b0e52f66e852d5fa781853e72ea1a8b4a4eea803b9731c2177f91969 -size 6426 +oid sha256:3fcf5557e3cffc6d17a57d1ee18134416f9542e1c71cb2fe007ff82a41e91850 +size 6669 diff --git a/tests/tensorflow/data/configs/resnet50_cifar10_magnitude_sparsity_int8.json b/tests/tensorflow/data/configs/resnet50_cifar10_magnitude_sparsity_int8.json index 83e4c48cd3e..c9c98284f9e 100644 --- a/tests/tensorflow/data/configs/resnet50_cifar10_magnitude_sparsity_int8.json +++ b/tests/tensorflow/data/configs/resnet50_cifar10_magnitude_sparsity_int8.json @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d1b649aca039431d0cd7cfba1ace2a4f310980f7dc2aebc9cc231218c3a8bbca -size 991 +oid sha256:937aceb493dfbc593f2a8e3371d0760fdbd66a0a09ff233c7835eb0b1213bf02 +size 1234 diff --git a/tests/tensorflow/data/configs/retinanet_coco2017_magnitude_sparsity_int8.json b/tests/tensorflow/data/configs/retinanet_coco2017_magnitude_sparsity_int8.json index 33d334bf97f..19217cb8e4c 100644 --- a/tests/tensorflow/data/configs/retinanet_coco2017_magnitude_sparsity_int8.json +++ b/tests/tensorflow/data/configs/retinanet_coco2017_magnitude_sparsity_int8.json @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:5978bb9d9a39176a2d74feb91fcd97682763d20b791cce08e10a988b8e321dc3 -size 2581 +oid sha256:79facd2934f9cf8f3e356e1c6f0c42a8330b70118a55491f57d6c5892542e7e1 +size 2824 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 index b5ad575aceb..51ebfe1f580 100644 --- a/tests/tensorflow/data/configs/sequential_model_cifar10_magnitude_sparsity_int8.json +++ b/tests/tensorflow/data/configs/sequential_model_cifar10_magnitude_sparsity_int8.json @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:3253a20090bf45eba3755216ef6d18b1308971c55a41d878af5d43f2aff8b07b -size 995 +oid sha256:db43fb5acf16522d9833b737a27b50c2807da7061caed7442c4444a824786ad8 +size 1238 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 index bb2e34ed709..8ade01ea744 100644 --- 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 @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:f92dfb5202beae5b8f4bbf64fd580fd9d22f03f66b502b61b92ca76f190953b2 -size 1002 +oid sha256:15d3d689db94a2413e40a1d46c55b4887abc991287f918b17022ae5fc5187b19 +size 1245 diff --git a/tests/tensorflow/test_sanity_sample.py b/tests/tensorflow/test_sanity_sample.py index cb1f43d4c2c..58afa5f9685 100644 --- a/tests/tensorflow/test_sanity_sample.py +++ b/tests/tensorflow/test_sanity_sample.py @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:f22d8dad705e1e0fceae937f5be460cdf6b5647bac3d70ee445a600b20f1e1f7 -size 15152 +oid sha256:fd6250f58209edcd80d3fbbcada80ddf4bcefededd542128618dff96b8fec6f9 +size 14420