Update calc_func.py

ultralytics · Jun 30, 2024 · 8a1de5e · 8a1de5e
1 parent cacd177
commit 8a1de5e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/thop/vision/calc_func.py b/thop/vision/calc_func.py
@@ -39,7 +39,6 @@ def calculate_conv2d_flops(input_size: list, output_size: list, kernel_size: lis
 def calculate_conv(bias, kernel_size, output_size, in_channel, group):
     """Calculate FLOPs for convolutional layers given bias, kernel size, output size, in_channels, and groups."""
     warnings.warn("This API is being deprecated.")
-    """Inputs are all numbers!"""
     return torch.DoubleTensor([output_size * (in_channel / group * kernel_size + bias)])