From 1e70fffb34c281581bfa3aee6f4c7365f5816385 Mon Sep 17 00:00:00 2001 From: shangliang Xu Date: Thu, 2 Jun 2022 14:06:04 +0800 Subject: [PATCH] [TIPC] add serving infer (#6116) * [TIPC] add serving python * [TIPC] alter amp_level=O2 --- deploy/serving/python/config.yml | 31 ++ deploy/serving/python/pipeline_http_client.py | 86 ++++ deploy/serving/python/preprocess_ops.py | 414 ++++++++++++++++++ deploy/serving/python/web_service.py | 247 +++++++++++ ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- 79 files changed, 853 insertions(+), 75 deletions(-) create mode 100644 deploy/serving/python/config.yml create mode 100644 deploy/serving/python/pipeline_http_client.py create mode 100644 deploy/serving/python/preprocess_ops.py create mode 100644 deploy/serving/python/web_service.py diff --git a/deploy/serving/python/config.yml b/deploy/serving/python/config.yml new file mode 100644 index 000000000..5ec428525 --- /dev/null +++ b/deploy/serving/python/config.yml @@ -0,0 +1,31 @@ +#worker_num, 最大并发数。当build_dag_each_worker=True时, 框架会创建worker_num个进程,每个进程内构建grpcSever和DAG +##当build_dag_each_worker=False时,框架会设置主线程grpc线程池的max_workers=worker_num +worker_num: 20 + +#http端口, rpc_port和http_port不允许同时为空。当rpc_port可用且http_port为空时,不自动生成http_port +http_port: 18093 +rpc_port: 9993 + +dag: + #op资源类型, True, 为线程模型;False,为进程模型 + is_thread_op: False +op: + #op名称,与web_service中的TIPCExampleService初始化name参数一致 + ppdet: + #并发数,is_thread_op=True时,为线程并发;否则为进程并发 + concurrency: 1 + + #当op配置没有server_endpoints时,从local_service_conf读取本地服务配置 + local_service_conf: + + #uci模型路径 + model_config: "./serving_server" + + #计算硬件类型: 空缺时由devices决定(CPU/GPU),0=cpu, 1=gpu, 2=tensorRT, 3=arm cpu, 4=kunlun xpu + device_type: + + #计算硬件ID,当devices为""或不写时为CPU预测;当devices为"0", "0,1,2"时为GPU预测,表示使用的GPU卡 + devices: "0" # "0,1" + + #client类型,包括brpc, grpc和local_predictor.local_predictor不启动Serving服务,进程内预测 + client_type: local_predictor diff --git a/deploy/serving/python/pipeline_http_client.py b/deploy/serving/python/pipeline_http_client.py new file mode 100644 index 000000000..33eb70dcf --- /dev/null +++ b/deploy/serving/python/pipeline_http_client.py @@ -0,0 +1,86 @@ +# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import glob +import requests +import json +import base64 +import os +import argparse + +parser = argparse.ArgumentParser(description="args for paddleserving") +parser.add_argument("--image_dir", type=str) +parser.add_argument("--image_file", type=str) +args = parser.parse_args() + + +def get_test_images(infer_dir, infer_img): + """ + Get image path list in TEST mode + """ + assert infer_img is not None or infer_dir is not None, \ + "--image_file or --image_dir should be set" + assert infer_img is None or os.path.isfile(infer_img), \ + "{} is not a file".format(infer_img) + assert infer_dir is None or os.path.isdir(infer_dir), \ + "{} is not a directory".format(infer_dir) + + # infer_img has a higher priority + if infer_img and os.path.isfile(infer_img): + return [infer_img] + + images = set() + infer_dir = os.path.abspath(infer_dir) + assert os.path.isdir(infer_dir), \ + "infer_dir {} is not a directory".format(infer_dir) + exts = ['jpg', 'jpeg', 'png', 'bmp'] + exts += [ext.upper() for ext in exts] + for ext in exts: + images.update(glob.glob('{}/*.{}'.format(infer_dir, ext))) + images = list(images) + + assert len(images) > 0, "no image found in {}".format(infer_dir) + print("Found {} inference images in total.".format(len(images))) + + return images + + +def cv2_to_base64(image): + """cv2_to_base64 + + Convert an numpy array to a base64 object. + + Args: + image: Input array. + + Returns: Base64 output of the input. + """ + return base64.b64encode(image).decode('utf8') + + +if __name__ == "__main__": + url = "http://127.0.0.1:18093/ppdet/prediction" + logid = 10000 + img_list = get_test_images(args.image_dir, args.image_file) + + for img_file in img_list: + with open(img_file, 'rb') as file: + image_data1 = file.read() + + image = cv2_to_base64(image_data1) + + data = {"key": ["image_0"], "value": [image], "logid": logid} + # send requests + r = requests.post(url=url, data=json.dumps(data)) + print(r.json()) diff --git a/deploy/serving/python/preprocess_ops.py b/deploy/serving/python/preprocess_ops.py new file mode 100644 index 000000000..eaf013744 --- /dev/null +++ b/deploy/serving/python/preprocess_ops.py @@ -0,0 +1,414 @@ +import numpy as np +import cv2 +import copy + + +def decode_image(im, img_info): + im = np.array(im) + img_info['im_shape'] = np.array(im.shape[:2], dtype=np.float32) + img_info['scale_factor'] = np.array([1., 1.], dtype=np.float32) + return im, img_info + + +class Resize(object): + """resize image by target_size and max_size + Args: + target_size (int): the target size of image + keep_ratio (bool): whether keep_ratio or not, default true + interp (int): method of resize + """ + + def __init__(self, target_size, keep_ratio=True, interp=cv2.INTER_LINEAR): + if isinstance(target_size, int): + target_size = [target_size, target_size] + self.target_size = target_size + self.keep_ratio = keep_ratio + self.interp = interp + + def __call__(self, im, im_info): + """ + Args: + im (np.ndarray): image (np.ndarray) + im_info (dict): info of image + Returns: + im (np.ndarray): processed image (np.ndarray) + im_info (dict): info of processed image + """ + assert len(self.target_size) == 2 + assert self.target_size[0] > 0 and self.target_size[1] > 0 + im_channel = im.shape[2] + im_scale_y, im_scale_x = self.generate_scale(im) + im = cv2.resize( + im, + None, + None, + fx=im_scale_x, + fy=im_scale_y, + interpolation=self.interp) + im_info['im_shape'] = np.array(im.shape[:2]).astype('float32') + im_info['scale_factor'] = np.array( + [im_scale_y, im_scale_x]).astype('float32') + return im, im_info + + def generate_scale(self, im): + """ + Args: + im (np.ndarray): image (np.ndarray) + Returns: + im_scale_x: the resize ratio of X + im_scale_y: the resize ratio of Y + """ + origin_shape = im.shape[:2] + im_c = im.shape[2] + if self.keep_ratio: + im_size_min = np.min(origin_shape) + im_size_max = np.max(origin_shape) + target_size_min = np.min(self.target_size) + target_size_max = np.max(self.target_size) + im_scale = float(target_size_min) / float(im_size_min) + if np.round(im_scale * im_size_max) > target_size_max: + im_scale = float(target_size_max) / float(im_size_max) + im_scale_x = im_scale + im_scale_y = im_scale + else: + resize_h, resize_w = self.target_size + im_scale_y = resize_h / float(origin_shape[0]) + im_scale_x = resize_w / float(origin_shape[1]) + return im_scale_y, im_scale_x + + +class NormalizeImage(object): + """normalize image + Args: + mean (list): im - mean + std (list): im / std + is_scale (bool): whether need im / 255 + is_channel_first (bool): if True: image shape is CHW, else: HWC + """ + + def __init__(self, mean, std, is_scale=True): + self.mean = mean + self.std = std + self.is_scale = is_scale + + def __call__(self, im, im_info): + """ + Args: + im (np.ndarray): image (np.ndarray) + im_info (dict): info of image + Returns: + im (np.ndarray): processed image (np.ndarray) + im_info (dict): info of processed image + """ + im = im.astype(np.float32, copy=False) + mean = np.array(self.mean)[np.newaxis, np.newaxis, :] + std = np.array(self.std)[np.newaxis, np.newaxis, :] + + if self.is_scale: + im = im / 255.0 + im -= mean + im /= std + return im, im_info + + +class Permute(object): + """permute image + Args: + to_bgr (bool): whether convert RGB to BGR + channel_first (bool): whether convert HWC to CHW + """ + + def __init__(self, ): + super(Permute, self).__init__() + + def __call__(self, im, im_info): + """ + Args: + im (np.ndarray): image (np.ndarray) + im_info (dict): info of image + Returns: + im (np.ndarray): processed image (np.ndarray) + im_info (dict): info of processed image + """ + im = im.transpose((2, 0, 1)).copy() + return im, im_info + + +class PadStride(object): + """ padding image for model with FPN, instead PadBatch(pad_to_stride) in original config + Args: + stride (bool): model with FPN need image shape % stride == 0 + """ + + def __init__(self, stride=0): + self.coarsest_stride = stride + + def __call__(self, im, im_info): + """ + Args: + im (np.ndarray): image (np.ndarray) + im_info (dict): info of image + Returns: + im (np.ndarray): processed image (np.ndarray) + im_info (dict): info of processed image + """ + coarsest_stride = self.coarsest_stride + if coarsest_stride <= 0: + return im, im_info + im_c, im_h, im_w = im.shape + pad_h = int(np.ceil(float(im_h) / coarsest_stride) * coarsest_stride) + pad_w = int(np.ceil(float(im_w) / coarsest_stride) * coarsest_stride) + padding_im = np.zeros((im_c, pad_h, pad_w), dtype=np.float32) + padding_im[:, :im_h, :im_w] = im + return padding_im, im_info + + +class LetterBoxResize(object): + def __init__(self, target_size): + """ + Resize image to target size, convert normalized xywh to pixel xyxy + format ([x_center, y_center, width, height] -> [x0, y0, x1, y1]). + Args: + target_size (int|list): image target size. + """ + super(LetterBoxResize, self).__init__() + if isinstance(target_size, int): + target_size = [target_size, target_size] + self.target_size = target_size + + def letterbox(self, img, height, width, color=(127.5, 127.5, 127.5)): + # letterbox: resize a rectangular image to a padded rectangular + shape = img.shape[:2] # [height, width] + ratio_h = float(height) / shape[0] + ratio_w = float(width) / shape[1] + ratio = min(ratio_h, ratio_w) + new_shape = (round(shape[1] * ratio), + round(shape[0] * ratio)) # [width, height] + padw = (width - new_shape[0]) / 2 + padh = (height - new_shape[1]) / 2 + top, bottom = round(padh - 0.1), round(padh + 0.1) + left, right = round(padw - 0.1), round(padw + 0.1) + + img = cv2.resize( + img, new_shape, interpolation=cv2.INTER_AREA) # resized, no border + img = cv2.copyMakeBorder( + img, top, bottom, left, right, cv2.BORDER_CONSTANT, + value=color) # padded rectangular + return img, ratio, padw, padh + + def __call__(self, im, im_info): + """ + Args: + im (np.ndarray): image (np.ndarray) + im_info (dict): info of image + Returns: + im (np.ndarray): processed image (np.ndarray) + im_info (dict): info of processed image + """ + assert len(self.target_size) == 2 + assert self.target_size[0] > 0 and self.target_size[1] > 0 + height, width = self.target_size + h, w = im.shape[:2] + im, ratio, padw, padh = self.letterbox(im, height=height, width=width) + + new_shape = [round(h * ratio), round(w * ratio)] + im_info['im_shape'] = np.array(new_shape, dtype=np.float32) + im_info['scale_factor'] = np.array([ratio, ratio], dtype=np.float32) + return im, im_info + + +class Pad(object): + def __init__(self, size, fill_value=[114.0, 114.0, 114.0]): + """ + Pad image to a specified size. + Args: + size (list[int]): image target size + fill_value (list[float]): rgb value of pad area, default (114.0, 114.0, 114.0) + """ + super(Pad, self).__init__() + if isinstance(size, int): + size = [size, size] + self.size = size + self.fill_value = fill_value + + def __call__(self, im, im_info): + im_h, im_w = im.shape[:2] + h, w = self.size + if h == im_h and w == im_w: + im = im.astype(np.float32) + return im, im_info + + canvas = np.ones((h, w, 3), dtype=np.float32) + canvas *= np.array(self.fill_value, dtype=np.float32) + canvas[0:im_h, 0:im_w, :] = im.astype(np.float32) + im = canvas + return im, im_info + + +def rotate_point(pt, angle_rad): + """Rotate a point by an angle. + + Args: + pt (list[float]): 2 dimensional point to be rotated + angle_rad (float): rotation angle by radian + + Returns: + list[float]: Rotated point. + """ + assert len(pt) == 2 + sn, cs = np.sin(angle_rad), np.cos(angle_rad) + new_x = pt[0] * cs - pt[1] * sn + new_y = pt[0] * sn + pt[1] * cs + rotated_pt = [new_x, new_y] + + return rotated_pt + + +def _get_3rd_point(a, b): + """To calculate the affine matrix, three pairs of points are required. This + function is used to get the 3rd point, given 2D points a & b. + + The 3rd point is defined by rotating vector `a - b` by 90 degrees + anticlockwise, using b as the rotation center. + + Args: + a (np.ndarray): point(x,y) + b (np.ndarray): point(x,y) + + Returns: + np.ndarray: The 3rd point. + """ + assert len(a) == 2 + assert len(b) == 2 + direction = a - b + third_pt = b + np.array([-direction[1], direction[0]], dtype=np.float32) + + return third_pt + + +def get_affine_transform(center, + input_size, + rot, + output_size, + shift=(0., 0.), + inv=False): + """Get the affine transform matrix, given the center/scale/rot/output_size. + + Args: + center (np.ndarray[2, ]): Center of the bounding box (x, y). + scale (np.ndarray[2, ]): Scale of the bounding box + wrt [width, height]. + rot (float): Rotation angle (degree). + output_size (np.ndarray[2, ]): Size of the destination heatmaps. + shift (0-100%): Shift translation ratio wrt the width/height. + Default (0., 0.). + inv (bool): Option to inverse the affine transform direction. + (inv=False: src->dst or inv=True: dst->src) + + Returns: + np.ndarray: The transform matrix. + """ + assert len(center) == 2 + assert len(output_size) == 2 + assert len(shift) == 2 + if not isinstance(input_size, (np.ndarray, list)): + input_size = np.array([input_size, input_size], dtype=np.float32) + scale_tmp = input_size + + shift = np.array(shift) + src_w = scale_tmp[0] + dst_w = output_size[0] + dst_h = output_size[1] + + rot_rad = np.pi * rot / 180 + src_dir = rotate_point([0., src_w * -0.5], rot_rad) + dst_dir = np.array([0., dst_w * -0.5]) + + src = np.zeros((3, 2), dtype=np.float32) + src[0, :] = center + scale_tmp * shift + src[1, :] = center + src_dir + scale_tmp * shift + src[2, :] = _get_3rd_point(src[0, :], src[1, :]) + + dst = np.zeros((3, 2), dtype=np.float32) + dst[0, :] = [dst_w * 0.5, dst_h * 0.5] + dst[1, :] = np.array([dst_w * 0.5, dst_h * 0.5]) + dst_dir + dst[2, :] = _get_3rd_point(dst[0, :], dst[1, :]) + + if inv: + trans = cv2.getAffineTransform(np.float32(dst), np.float32(src)) + else: + trans = cv2.getAffineTransform(np.float32(src), np.float32(dst)) + + return trans + + +class WarpAffine(object): + """Warp affine the image + """ + + def __init__(self, + keep_res=False, + pad=31, + input_h=512, + input_w=512, + scale=0.4, + shift=0.1): + self.keep_res = keep_res + self.pad = pad + self.input_h = input_h + self.input_w = input_w + self.scale = scale + self.shift = shift + + def __call__(self, im, im_info): + """ + Args: + im (np.ndarray): image (np.ndarray) + im_info (dict): info of image + Returns: + im (np.ndarray): processed image (np.ndarray) + im_info (dict): info of processed image + """ + img = cv2.cvtColor(im, cv2.COLOR_RGB2BGR) + + h, w = img.shape[:2] + + if self.keep_res: + input_h = (h | self.pad) + 1 + input_w = (w | self.pad) + 1 + s = np.array([input_w, input_h], dtype=np.float32) + c = np.array([w // 2, h // 2], dtype=np.float32) + + else: + s = max(h, w) * 1.0 + input_h, input_w = self.input_h, self.input_w + c = np.array([w / 2., h / 2.], dtype=np.float32) + + trans_input = get_affine_transform(c, s, 0, [input_w, input_h]) + img = cv2.resize(img, (w, h)) + inp = cv2.warpAffine( + img, trans_input, (input_w, input_h), flags=cv2.INTER_LINEAR) + return inp, im_info + + +class Compose: + def __init__(self, transforms): + self.transforms = [] + for op_info in transforms: + new_op_info = op_info.copy() + op_type = new_op_info.pop('type') + self.transforms.append(eval(op_type)(**new_op_info)) + + self.im_info = { + 'scale_factor': np.array( + [1., 1.], dtype=np.float32), + 'im_shape': None + } + + def __call__(self, img): + img, self.im_info = decode_image(img, self.im_info) + for t in self.transforms: + img, self.im_info = t(img, self.im_info) + inputs = copy.deepcopy(self.im_info) + inputs['image'] = img + return inputs diff --git a/deploy/serving/python/web_service.py b/deploy/serving/python/web_service.py new file mode 100644 index 000000000..5a7018f98 --- /dev/null +++ b/deploy/serving/python/web_service.py @@ -0,0 +1,247 @@ +# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +import copy + +from paddle_serving_server.web_service import WebService, Op +from paddle_serving_server.proto import general_model_config_pb2 as m_config +import google.protobuf.text_format + +import os +import numpy as np +import base64 +from PIL import Image +import io +from preprocess_ops import Compose + +from argparse import ArgumentParser, RawDescriptionHelpFormatter +import yaml + +# Global dictionary +SUPPORT_MODELS = { + 'YOLO', + 'RCNN', + 'SSD', + 'Face', + 'FCOS', + 'SOLOv2', + 'TTFNet', + 'S2ANet', + 'JDE', + 'FairMOT', + 'DeepSORT', + 'GFL', + 'PicoDet', + 'CenterNet', + 'TOOD', + 'RetinaNet', + 'StrongBaseline', + 'STGCN', + 'YOLOX', +} + +GLOBAL_VAR = {} + + +class ArgsParser(ArgumentParser): + def __init__(self): + super(ArgsParser, self).__init__( + formatter_class=RawDescriptionHelpFormatter) + self.add_argument( + "-c", + "--config", + default="deploy/serving/python/config.yml", + help="configuration file to use") + self.add_argument( + "--model_dir", + type=str, + default=None, + help=("Directory include:'model.pdiparams', 'model.pdmodel', " + "'infer_cfg.yml', created by tools/export_model.py."), + required=True) + self.add_argument( + "-o", "--opt", nargs='+', help="set configuration options") + + def parse_args(self, argv=None): + args = super(ArgsParser, self).parse_args(argv) + assert args.config is not None, \ + "Please specify --config=configure_file_path." + args.service_config = self._parse_opt(args.opt, args.config) + print("args config:", args.service_config) + args.model_config = PredictConfig(args.model_dir) + return args + + def _parse_helper(self, v): + if v.isnumeric(): + if "." in v: + v = float(v) + else: + v = int(v) + elif v == "True" or v == "False": + v = (v == "True") + return v + + def _parse_opt(self, opts, conf_path): + f = open(conf_path) + config = yaml.load(f, Loader=yaml.Loader) + if not opts: + return config + for s in opts: + s = s.strip() + k, v = s.split('=') + v = self._parse_helper(v) + if "devices" in k: + v = str(v) + print(k, v, type(v)) + cur = config + parent = cur + for kk in k.split("."): + if kk not in cur: + cur[kk] = {} + parent = cur + cur = cur[kk] + else: + parent = cur + cur = cur[kk] + parent[k.split(".")[-1]] = v + return config + + +class PredictConfig(object): + """set config of preprocess, postprocess and visualize + Args: + model_dir (str): root path of infer_cfg.yml + """ + + def __init__(self, model_dir): + # parsing Yaml config for Preprocess + deploy_file = os.path.join(model_dir, 'infer_cfg.yml') + with open(deploy_file) as f: + yml_conf = yaml.safe_load(f) + self.check_model(yml_conf) + self.arch = yml_conf['arch'] + self.preprocess_infos = yml_conf['Preprocess'] + self.min_subgraph_size = yml_conf['min_subgraph_size'] + self.labels = yml_conf['label_list'] + self.use_dynamic_shape = yml_conf['use_dynamic_shape'] + self.draw_threshold = yml_conf.get("draw_threshold", 0.5) + self.mask = yml_conf.get("mask", False) + self.tracker = yml_conf.get("tracker", None) + self.nms = yml_conf.get("NMS", None) + self.fpn_stride = yml_conf.get("fpn_stride", None) + if self.arch == 'RCNN' and yml_conf.get('export_onnx', False): + print( + 'The RCNN export model is used for ONNX and it only supports batch_size = 1' + ) + self.print_config() + + def check_model(self, yml_conf): + """ + Raises: + ValueError: loaded model not in supported model type + """ + for support_model in SUPPORT_MODELS: + if support_model in yml_conf['arch']: + return True + raise ValueError("Unsupported arch: {}, expect {}".format(yml_conf[ + 'arch'], SUPPORT_MODELS)) + + def print_config(self): + print('----------- Model Configuration -----------') + print('%s: %s' % ('Model Arch', self.arch)) + print('%s: ' % ('Transform Order')) + for op_info in self.preprocess_infos: + print('--%s: %s' % ('transform op', op_info['type'])) + print('--------------------------------------------') + + +class DetectorOp(Op): + def init_op(self): + self.preprocess_pipeline = Compose(GLOBAL_VAR['preprocess_ops']) + + def preprocess(self, input_dicts, data_id, log_id): + (_, input_dict), = input_dicts.items() + inputs = [] + for key, data in input_dict.items(): + data = base64.b64decode(data.encode('utf8')) + byte_stream = io.BytesIO(data) + img = Image.open(byte_stream).convert("RGB") + inputs.append(self.preprocess_pipeline(img)) + inputs = self.collate_inputs(inputs) + return inputs, False, None, "" + + def postprocess(self, input_dicts, fetch_dict, data_id, log_id): + (_, input_dict), = input_dicts.items() + bboxes = fetch_dict["multiclass_nms3_0.tmp_0"] + bboxes_num = fetch_dict["multiclass_nms3_0.tmp_2"] + draw_threshold = GLOBAL_VAR['model_config'].draw_threshold + idx = 0 + result = {} + for k, num in zip(input_dict.keys(), bboxes_num): + bbox = bboxes[idx:idx + num] + result[k] = self.parse_det_result(bbox, draw_threshold, + GLOBAL_VAR['model_config'].labels) + return result, None, "" + + def collate_inputs(self, inputs): + collate_inputs = {k: [] for k in inputs[0].keys()} + for info in inputs: + for k in collate_inputs.keys(): + collate_inputs[k].append(info[k]) + return { + k: np.stack(v) + for k, v in collate_inputs.items() if k in GLOBAL_VAR['feed_vars'] + } + + def parse_det_result(self, bbox, draw_threshold, label_list): + result = [] + for line in bbox: + if line[1] > draw_threshold: + result.append(f"{label_list[int(line[0])]} {line[1]} " + f"{line[2]} {line[3]} {line[4]} {line[5]}") + return result + + +class DetectorService(WebService): + def get_pipeline_response(self, read_op): + return DetectorOp(name="ppdet", input_ops=[read_op]) + + +def get_model_vars(model_dir, service_config): + serving_server_dir = os.path.join(model_dir, "serving_server") + # rewrite model_config + service_config['op']['ppdet']['local_service_conf'][ + 'model_config'] = serving_server_dir + f = open( + os.path.join(serving_server_dir, "serving_server_conf.prototxt"), 'r') + model_var = google.protobuf.text_format.Merge( + str(f.read()), m_config.GeneralModelConfig()) + feed_vars = [var.name for var in model_var.feed_var] + fetch_vars = [var.name for var in model_var.fetch_var] + return feed_vars, fetch_vars + + +if __name__ == '__main__': + # load config and prepare the service + FLAGS = ArgsParser().parse_args() + feed_vars, fetch_vars = get_model_vars(FLAGS.model_dir, + FLAGS.service_config) + GLOBAL_VAR['feed_vars'] = feed_vars + GLOBAL_VAR['fetch_vars'] = fetch_vars + GLOBAL_VAR['preprocess_ops'] = FLAGS.model_config.preprocess_infos + GLOBAL_VAR['model_config'] = FLAGS.model_config + # define the service + uci_service = DetectorService(name="ppdet") + uci_service.prepare_pipeline_config(yml_dict=FLAGS.service_config) + # start the service + uci_service.run_service() diff --git a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index e9bc4e640..5a38b046f 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/cascade_mask_rcnn_r50_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 446bcd0b1..5d49ea4bd 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml -o diff --git a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 4ce15810e..f1c3e3f37 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/cascade_rcnn_r50_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index c9c013025..277437f29 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml -o diff --git a/test_tipc/configs/deformable_detr/deformable_detr_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/deformable_detr/deformable_detr_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 7ac82f945..49821ac53 100644 --- a/test_tipc/configs/deformable_detr/deformable_detr_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/deformable_detr/deformable_detr_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/deformable_detr_r50_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/deformable_detr/deformable_detr_r50_1x_coco.yml -o diff --git a/test_tipc/configs/detr/detr_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/detr/detr_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 4b49de9da..b6073456f 100644 --- a/test_tipc/configs/detr/detr_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/detr/detr_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/detr_r50_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/detr/detr_r50_1x_coco.yml -o diff --git a/test_tipc/configs/dota/s2anet_1x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/dota/s2anet_1x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 287c53afa..08f29765f 100644 --- a/test_tipc/configs/dota/s2anet_1x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/dota/s2anet_1x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/s2anet_1x_spine.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/dota/s2anet_1x_spine.yml -o diff --git a/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 00475ef3e..04b2082c5 100644 --- a/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/s2anet_alignconv_2x_dota.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/spine_coco/test/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml -o diff --git a/test_tipc/configs/dota/s2anet_conv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/dota/s2anet_conv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 8f2d0992d..f88d2241a 100644 --- a/test_tipc/configs/dota/s2anet_conv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/dota/s2anet_conv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/s2anet_conv_2x_dota.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/spine_coco/test/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml -o diff --git a/test_tipc/configs/face_detection/blazeface_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/face_detection/blazeface_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index a4d45d52e..1300f81d8 100644 --- a/test_tipc/configs/face_detection/blazeface_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/face_detection/blazeface_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/blazeface_1000e.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/face_detection/blazeface_1000e.yml -o diff --git a/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index c654b6de5..b80b650d9 100644 --- a/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/blazeface_fpn_ssh_1000e.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 4d824f157..e2c96858a 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 56406cc53..e6ef4154e 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 6423f3afc..0eb12969a 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_vd_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index fcc33579b..a7ea74d43 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r34_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 57a10cb86..9c45cd772 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r34_vd_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 5ce75c465..a1584b90e 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index d886e94af..da2470ac2 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 4c6cc960c..c12efa6cd 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_vd_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index a8ca7c610..a852290aa 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_vd_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index ef1a06f00..6fc1c6edc 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_swin_tiny_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 7cfcac208..9dda811bb 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/fcos/fcos_dcn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/fcos/fcos_dcn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 2320b41f8..88bc7aa08 100644 --- a/test_tipc/configs/fcos/fcos_dcn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/fcos/fcos_dcn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/fcos_dcn_r50_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/fcos/fcos_dcn_r50_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 420417f02..6475ac886 100644 --- a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/fcos_r50_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/fcos/fcos_r50_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/gfl/gfl_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/gfl/gfl_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index d491ec0ca..22169af14 100644 --- a/test_tipc/configs/gfl/gfl_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/gfl/gfl_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/gfl_r50_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/gfl/gfl_r50_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 8b334240c..787505406 100644 --- a/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/keypoint/dark_hrnet_w32_256x192.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/keypoint/hrnet/dark_hrnet_w32_256x192.yml -o diff --git a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index b50aa8f9c..8a6c6f17a 100644 --- a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/keypoint/dark_hrnet_w48_256x192.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml -o diff --git a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index c3496b41d..9b5e0503e 100644 --- a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/keypoint/higherhrnet_hrnet_w32_512.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/keypoint/higherhrnet/higherhrnet_hrnet_w32_512.yml -o diff --git a/test_tipc/configs/keypoint/hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 9f8aaf54f..26c6a3df0 100644 --- a/test_tipc/configs/keypoint/hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/keypoint/hrnet_w32_256x192.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/keypoint/hrnet/hrnet_w32_256x192.yml -o diff --git a/test_tipc/configs/keypoint/tinypose_128x96_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/tinypose_128x96_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index e0198b289..981118952 100644 --- a/test_tipc/configs/keypoint/tinypose_128x96_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/tinypose_128x96_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/keypoint/tinypose_128x96.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index cebdb83fb..2be777af8 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mask_rcnn_r101_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 0d7f36ab6..8e26f3443 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mask_rcnn_r101_vd_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index ede384570..843ac40e9 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml -o diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 46447daad..af061e055 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 56d3e2612..c2cc1429e 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_vd_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 705f0b3ed..d2c226a1f 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/mot/fairmot_dla34_30e_1088x608_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mot/fairmot_dla34_30e_1088x608_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index db624ce64..914b74175 100644 --- a/test_tipc/configs/mot/fairmot_dla34_30e_1088x608_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mot/fairmot_dla34_30e_1088x608_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/mot/test.mp4 -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml -o diff --git a/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 0de7ed6ce..2cb87aaa2 100644 --- a/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/mot/test.mp4 -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml -o diff --git a/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index bc4e37e68..f4407bd75 100644 --- a/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/mot/test.mp4 -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml -o diff --git a/test_tipc/configs/mot/jde_darknet53_30e_1088x608_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mot/jde_darknet53_30e_1088x608_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 7decaedc0..45b8d17c5 100644 --- a/test_tipc/configs/mot/jde_darknet53_30e_1088x608_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mot/jde_darknet53_30e_1088x608_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/jde_darknet53_30e_1088x608.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/mot/test.mp4 -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mot/jde/jde_darknet53_30e_1088x608.yml -o diff --git a/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 18a0549f5..40d98490f 100644 --- a/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/jde_darknet53_30e_576x320.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/mot/test.mp4 -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml -o diff --git a/test_tipc/configs/picodet/picodet_l_640_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_l_640_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index a2aa277cf..4fb5661f5 100644 --- a/test_tipc/configs/picodet/picodet_l_640_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_l_640_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_l_640_coco_lcnet.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/picodet_l_640_coco_lcnet.yml -o diff --git a/test_tipc/configs/picodet/picodet_l_640_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_l_640_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index c8ba16da1..6b54d6543 100644 --- a/test_tipc/configs/picodet/picodet_l_640_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_l_640_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_l_640_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/legacy_model/picodet_l_640_coco.yml -o diff --git a/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index ba0d31ccf..28272ad58 100644 --- a/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_lcnet_1_5x_416_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/legacy_model/more_config/picodet_lcnet_1_5x_416_coco.yml -o diff --git a/test_tipc/configs/picodet/picodet_m_416_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_m_416_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 200c0e448..b347c375f 100644 --- a/test_tipc/configs/picodet/picodet_m_416_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_m_416_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_m_416_coco_lcnet.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/picodet_m_416_coco_lcnet.yml -o diff --git a/test_tipc/configs/picodet/picodet_m_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_m_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 5e1507633..8d99e83d6 100644 --- a/test_tipc/configs/picodet/picodet_m_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_m_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_m_416_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/legacy_model/picodet_m_416_coco.yml -o diff --git a/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index e9fe5361c..3db11e54f 100644 --- a/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_mobilenetv3_large_1x_416_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/legacy_model/more_config/picodet_mobilenetv3_large_1x_416_coco.yml -o diff --git a/test_tipc/configs/picodet/picodet_r18_640_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_r18_640_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 87997a3a5..8a5e48793 100644 --- a/test_tipc/configs/picodet/picodet_r18_640_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_r18_640_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_r18_640_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/legacy_model/more_config/picodet_r18_640_coco.yml -o diff --git a/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 776d02ed4..cac5a70bb 100644 --- a/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_s_320_coco_lcnet.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/picodet_s_320_coco_lcnet.yml -o diff --git a/test_tipc/configs/picodet/picodet_s_320_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_s_320_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 5765a1458..b1a87cdd7 100644 --- a/test_tipc/configs/picodet/picodet_s_320_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_s_320_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_s_320_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/legacy_model/picodet_s_320_coco.yml -o diff --git a/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 72cc4dc0d..065abae66 100644 --- a/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_shufflenetv2_1x_416_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/legacy_model/more_config/picodet_shufflenetv2_1x_416_coco.yml -o diff --git a/test_tipc/configs/picodet/picodet_xs_320_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/picodet/picodet_xs_320_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 89934723b..c84d14c38 100644 --- a/test_tipc/configs/picodet/picodet_xs_320_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/picodet/picodet_xs_320_coco_lcnet_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/picodet_xs_320_coco_lcnet.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/picodet/picodet_xs_320_coco_lcnet.yml -o diff --git a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index ea56d0c6e..6ee39190d 100644 --- a/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyolo/ppyolo_mbv3_large_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyolo_mbv3_large_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyolo/ppyolo_mbv3_large_coco.yml -o diff --git a/test_tipc/configs/ppyolo/ppyolo_mbv3_small_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyolo/ppyolo_mbv3_small_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 75a1903d5..ae3322eee 100644 --- a/test_tipc/configs/ppyolo/ppyolo_mbv3_small_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyolo/ppyolo_mbv3_small_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyolo_mbv3_small_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyolo/ppyolo_mbv3_small_coco.yml -o diff --git a/test_tipc/configs/ppyolo/ppyolo_r18vd_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyolo/ppyolo_r18vd_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 1adeaf4b4..fd2ef0dca 100644 --- a/test_tipc/configs/ppyolo/ppyolo_r18vd_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyolo/ppyolo_r18vd_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyolo_r18vd_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyolo/ppyolo_r18vd_coco.yml -o diff --git a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 3208f48ff..6a4235a0b 100644 --- a/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyolo/ppyolo_r50vd_dcn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyolo_r50vd_dcn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyolo/ppyolo_r50vd_dcn_1x_coco.yml -o diff --git a/test_tipc/configs/ppyolo/ppyolo_tiny_650e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyolo/ppyolo_tiny_650e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index caabd60b3..cb0659a7b 100644 --- a/test_tipc/configs/ppyolo/ppyolo_tiny_650e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyolo/ppyolo_tiny_650e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyolo_tiny_650e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyolo/ppyolo_tiny_650e_coco.yml -o diff --git a/test_tipc/configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index d42bdbaaa..fc2ca6768 100644 --- a/test_tipc/configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyolov2_r101vd_dcn_365e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco.yml -o diff --git a/test_tipc/configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 3c520c9e0..721db09e0 100644 --- a/test_tipc/configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyolov2_r50vd_dcn_365e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco.yml -o diff --git a/test_tipc/configs/ppyoloe/ppyoloe_crn_l_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyoloe/ppyoloe_crn_l_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index b6f75f7bd..3df9e020b 100644 --- a/test_tipc/configs/ppyoloe/ppyoloe_crn_l_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyoloe/ppyoloe_crn_l_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyoloe_crn_l_300e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyoloe/ppyoloe_crn_l_300e_coco.yml -o diff --git a/test_tipc/configs/ppyoloe/ppyoloe_crn_m_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyoloe/ppyoloe_crn_m_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index aafe6a093..607113b9d 100644 --- a/test_tipc/configs/ppyoloe/ppyoloe_crn_m_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyoloe/ppyoloe_crn_m_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyoloe_crn_m_300e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyoloe/ppyoloe_crn_m_300e_coco.yml -o diff --git a/test_tipc/configs/ppyoloe/ppyoloe_crn_s_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyoloe/ppyoloe_crn_s_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 5ea846972..815a23937 100644 --- a/test_tipc/configs/ppyoloe/ppyoloe_crn_s_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyoloe/ppyoloe_crn_s_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyoloe_crn_s_300e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyoloe/ppyoloe_crn_s_300e_coco.yml -o diff --git a/test_tipc/configs/ppyoloe/ppyoloe_crn_x_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ppyoloe/ppyoloe_crn_x_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 7ed12a7a8..38f2fcc8f 100644 --- a/test_tipc/configs/ppyoloe/ppyoloe_crn_x_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ppyoloe/ppyoloe_crn_x_300e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ppyoloe_crn_x_300e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ppyoloe/ppyoloe_crn_x_300e_coco.yml -o diff --git a/test_tipc/configs/retinanet/retinanet_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/retinanet/retinanet_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 76a6b9e57..585c99f38 100644 --- a/test_tipc/configs/retinanet/retinanet_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/retinanet/retinanet_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/retinanet_r50_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/retinanet/retinanet_r50_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 142d03543..df50fd3c4 100644 --- a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/solov2_r101_vd_fpn_3x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/solov2/solov2_r101_vd_fpn_3x_coco.yml -o diff --git a/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 4195c11a4..e2b43e45f 100644 --- a/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/solov2_r50_enhance_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/solov2/solov2_r50_enhance_coco.yml -o diff --git a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 41ed5ef46..3d0050402 100644 --- a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/solov2_r50_fpn_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/solov2/solov2_r50_fpn_1x_coco.yml -o diff --git a/test_tipc/configs/sparse_rcnn/sparse_rcnn_r50_fpn_3x_pro100_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/sparse_rcnn/sparse_rcnn_r50_fpn_3x_pro100_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index f4dbf76f5..a461713b3 100644 --- a/test_tipc/configs/sparse_rcnn/sparse_rcnn_r50_fpn_3x_pro100_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/sparse_rcnn/sparse_rcnn_r50_fpn_3x_pro100_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/sparse_rcnn_r50_fpn_3x_pro100_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/sparse_rcnn/sparse_rcnn_r50_fpn_3x_pro100_coco.yml -o diff --git a/test_tipc/configs/ssd/ssdlite_mobilenet_v1_300_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ssd/ssdlite_mobilenet_v1_300_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index ec50a05a8..8d2036f55 100644 --- a/test_tipc/configs/ssd/ssdlite_mobilenet_v1_300_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ssd/ssdlite_mobilenet_v1_300_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ssdlite_mobilenet_v1_300_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ssd/ssdlite_mobilenet_v1_300_coco.yml -o diff --git a/test_tipc/configs/ttfnet/ttfnet_darknet53_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/ttfnet/ttfnet_darknet53_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 1d33fe812..a167ce93d 100644 --- a/test_tipc/configs/ttfnet/ttfnet_darknet53_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/ttfnet/ttfnet_darknet53_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/ttfnet_darknet53_1x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/ttfnet/ttfnet_darknet53_1x_coco.yml -o diff --git a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index cfe8534bb..b3142b6ea 100644 --- a/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/yolov3/yolov3_darknet53_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/yolov3_darknet53_270e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml -o diff --git a/test_tipc/configs/yolov3/yolov3_mobilenet_v1_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/yolov3/yolov3_mobilenet_v1_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 6f9cb1424..5e9d35f2a 100644 --- a/test_tipc/configs/yolov3/yolov3_mobilenet_v1_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/yolov3/yolov3_mobilenet_v1_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_270e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/yolov3/yolov3_mobilenet_v1_270e_coco.yml -o diff --git a/test_tipc/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index be099dd9d..49668ed6c 100644 --- a/test_tipc/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_270e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml -o diff --git a/test_tipc/configs/yolov3/yolov3_r34_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/yolov3/yolov3_r34_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 667b6c865..6b8720cdc 100644 --- a/test_tipc/configs/yolov3/yolov3_r34_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/yolov3/yolov3_r34_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/yolov3_r34_270e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/yolov3/yolov3_r34_270e_coco.yml -o diff --git a/test_tipc/configs/yolov3/yolov3_r50vd_dcn_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/yolov3/yolov3_r50vd_dcn_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index a1d20bab8..4282bb38c 100644 --- a/test_tipc/configs/yolov3/yolov3_r50vd_dcn_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/yolov3/yolov3_r50vd_dcn_270e_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -10,7 +10,7 @@ TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_tr pretrain_weights:https://paddledet.bj.bcebos.com/models/yolov3_r50vd_dcn_270e_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ -filename:null +amp_level:O2 ## trainer:norm_train norm_train:tools/train.py -c configs/yolov3/yolov3_r50vd_dcn_270e_coco.yml -o -- GitLab