|
|
|
@ -136,7 +136,7 @@ class TestCalibrationForResnet50(unittest.TestCase):
|
|
|
|
|
"full_data", False)
|
|
|
|
|
else:
|
|
|
|
|
data_urls.append(
|
|
|
|
|
'http://paddle-inference-dist.cdn.bcebos.com/int8/calibration_test_data.tar.gz'
|
|
|
|
|
'http://paddle-inference-dist.bj.bcebos.com/int8/calibration_test_data.tar.gz'
|
|
|
|
|
)
|
|
|
|
|
data_md5s.append('1b6c1c434172cca1bf9ba1e4d7a3157d')
|
|
|
|
|
self.data_cache_folder = self.download_data(data_urls, data_md5s,
|
|
|
|
@ -189,7 +189,7 @@ class TestCalibrationForResnet50(unittest.TestCase):
|
|
|
|
|
def download_model(self):
|
|
|
|
|
# resnet50 fp32 data
|
|
|
|
|
data_urls = [
|
|
|
|
|
'http://paddle-inference-dist.cdn.bcebos.com/int8/resnet50_int8_model.tar.gz'
|
|
|
|
|
'http://paddle-inference-dist.bj.bcebos.com/int8/resnet50_int8_model.tar.gz'
|
|
|
|
|
]
|
|
|
|
|
data_md5s = ['4a5194524823d9b76da6e738e1367881']
|
|
|
|
|
self.model_cache_folder = self.download_data(data_urls, data_md5s,
|
|
|
|
@ -307,7 +307,7 @@ class TestCalibrationForMobilenetv1(TestCalibrationForResnet50):
|
|
|
|
|
def download_model(self):
|
|
|
|
|
# mobilenetv1 fp32 data
|
|
|
|
|
data_urls = [
|
|
|
|
|
'http://paddle-inference-dist.cdn.bcebos.com/int8/mobilenetv1_int8_model.tar.gz'
|
|
|
|
|
'http://paddle-inference-dist.bj.bcebos.com/int8/mobilenetv1_int8_model.tar.gz'
|
|
|
|
|
]
|
|
|
|
|
data_md5s = ['13892b0716d26443a8cdea15b3c6438b']
|
|
|
|
|
self.model_cache_folder = self.download_data(data_urls, data_md5s,
|
|
|
|
|