From 0c9f9b94732bbbff36a3efb725611f4c4a42b96e Mon Sep 17 00:00:00 2001
From: Anthony Dave <davidietop@163.com>
Date: Tue, 14 Nov 2023 20:30:07 +0800
Subject: [PATCH] import weight_init class for LastLevelP6

---
 swint/swin_transformer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/swint/swin_transformer.py b/swint/swin_transformer.py
index 64813f6..95fb7bc 100644
--- a/swint/swin_transformer.py
+++ b/swint/swin_transformer.py
@@ -7,7 +7,9 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.utils.checkpoint as checkpoint
+
 import numpy as np
+import fvcore.nn.weight_init as weight_init
 from timm.models.layers import DropPath, to_2tuple, trunc_normal_
 
 from detectron2.modeling.backbone import Backbone