datasets.py self.batch_shapes fix

This commit is contained in:
Glenn Jocher committed 2020-06-01 18:11:18 -07:00
1 parent 7a00a5eff2
commit ee8988b8a2
3 files changed
+3 -4

No files matched your search

-1
View File
@@ -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):