datasets.py self.batch_shapes fix
This commit is contained in:
1 parent
7a00a5eff2
commit
ee8988b8a2
3 files changed
+3
-4
No files matched your search
@@ -20,7 +20,6 @@ class Detect(nn.Module):
|
||||
self.export = False # onnx export
|
||||
|
||||
def forward(self, x):
|
||||
x = x.copy()
|
||||
z = [] # inference output
|
||||
self.training |= self.export
|
||||
for i in range(self.nl):
|
||||
|
||||
Reference in new issue
Block a user