From 85d7ae21959a09028bc3becf15bd918450d92ad8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 26 Aug 2022 14:34:28 +0200 Subject: [PATCH] Rename onnx_dynamic -> dynamic (#9168) --- export.py | 2 +- models/yolo.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/export.py b/export.py index 1bb7ded..0f26e63 100644 --- a/export.py +++ b/export.py @@ -489,7 +489,7 @@ def run( for k, m in model.named_modules(): if isinstance(m, Detect): m.inplace = inplace - m.onnx_dynamic = dynamic + m.dynamic = dynamic m.export = True for _ in range(2): diff --git a/models/yolo.py b/models/yolo.py index e154b72..7a73083 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -37,7 +37,7 @@ except ImportError: class Detect(nn.Module): stride = None # strides computed during build - onnx_dynamic = False # ONNX export parameter + dynamic = False # force grid reconstruction export = False # export mode def __init__(self, nc=80, anchors=(), ch=(), inplace=True): # detection layer @@ -60,7 +60,7 @@ class Detect(nn.Module): x[i] = x[i].view(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous() if not self.training: # inference - if self.onnx_dynamic or self.grid[i].shape[2:4] != x[i].shape[2:4]: + if self.dynamic or self.grid[i].shape[2:4] != x[i].shape[2:4]: self.grid[i], self.anchor_grid[i] = self._make_grid(nx, ny, i) y = x[i].sigmoid()