def test_nncf_compress_and_export(self):
log_file, metrics_path = super().test_nncf_compress_and_export()
compress_ap = collect_ap(log_file)
last_compress_ap = compress_ap[-1]
logging.info(f'From training last_compress_ap={last_compress_ap}')
ap = self._get_bbox_metric(metrics_path)
logging.info(f'Evaluation after export result ap={ap}')
self.assertGreater(ap, last_compress_ap - self.test_export_thr)
def test_nncf_compress_and_eval_on_gpu(self):
log_file, metrics_path = super(
).test_nncf_compress_and_eval_on_gpu()
compress_ap = collect_ap(log_file)
last_compress_ap = compress_ap[-1]
logging.info(f'From training last_compress_ap={last_compress_ap}')
ap = self._get_bbox_metric(metrics_path)
logging.info(f'Evaluation result ap={ap}')
self.assertLess(abs(last_compress_ap - ap), 1e-6)
return log_file, metrics_path
def test_nncf_compress_on_gpu(self):
skip_if_cuda_not_available()
log_file = super().test_nncf_compress_on_gpu()
ap = collect_ap(log_file)
self.assertGreater(ap[-1], 0)
def test_nncf_compress_on_gpu(self):
log_file = super().test_nncf_compress_on_gpu()
ap = collect_ap(log_file)
self.assertGreater(ap[-1], 0)