From 91af0401d8c3684903a07b1a14c2231d75971c61 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 28 Jul 2020 14:48:33 -0700
Subject: [PATCH] PyTorch 1.6.0 compatibility updates

---
 models/yolo.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/models/yolo.py b/models/yolo.py
index 47a35e2..21923fa 100644
--- a/models/yolo.py
+++ b/models/yolo.py
@@ -148,6 +148,7 @@ class Model(nn.Module):
         print('Fusing layers... ', end='')
         for m in self.model.modules():
             if type(m) is Conv:
+                m._non_persistent_buffers_set = set()  # pytorch 1.6.0 compatability
                 m.conv = torch_utils.fuse_conv_and_bn(m.conv, m.bn)  # update conv
                 m.bn = None  # remove batchnorm
                 m.forward = m.fuseforward  # update forward