import weight_init class for LastLevelP6

xiaohu2015 · Nov 14, 2023 · 0c9f9b9 · 0c9f9b9
1 parent d1d637e
commit 0c9f9b9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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