Update EFEN_arch.py

Supereeeee · Mar 5, 2024 · cacfa46 · cacfa46
1 parent 33355aa
commit cacfa46
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/basicsr/archs/EFEN_arch.py b/basicsr/archs/EFEN_arch.py
@@ -6,8 +6,8 @@
 from torch import nn as nn
 from torch.nn import functional as F
 from basicsr.utils.registry import ARCH_REGISTRY
-from .arch_util import default_init_weights, make_layer
-from thop import profile  # 计算参数量和运算量
+from .arch_util import make_layer
+from thop import profile
 
 class BSConvU(torch.nn.Sequential):
     def __init__(self, in_channels, out_channels, kernel_size, padding=1, stride=1, dilation=1, bias=True,