W0510 00:32:22.100802 8743 gpu_context.cc:244] Please NOTE: device: 0, GPU Compute Capability: 7.0, Driver API Version: 11.2, Runtime API Version: 10.1 W0510 00:32:22.104816 8743 gpu_context.cc:272] device: 0, cuDNN Version: 7.6.
------------------------------------------------------------------------------- Layer (type) Input Shape Output Shape Param # =============================================================================== Conv2D-1 [[1, 3, 256, 256]] [1, 64, 128, 128] 9,408 BatchNorm-1 [[1, 64, 128, 128]] [1, 64, 128, 128] 256 ConvBNLayer-1 [[1, 3, 256, 256]] [1, 64, 128, 128] 0 MaxPool2D-1 [[1, 64, 128, 128]] [1, 64, 64, 64] 0 Conv2D-2 [[1, 64, 64, 64]] [1, 256, 64, 64] 16,384 BatchNorm-2 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-2 [[1, 64, 64, 64]] [1, 256, 64, 64] 0 Conv2D-3 [[1, 256, 64, 64]] [1, 256, 64, 64] 9,216 BatchNorm-3 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-3 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 Conv2D-4 [[1, 256, 64, 64]] [1, 256, 64, 64] 65,536 BatchNorm-4 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-4 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 Conv2D-5 [[1, 64, 64, 64]] [1, 256, 64, 64] 16,384 BatchNorm-5 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-5 [[1, 64, 64, 64]] [1, 256, 64, 64] 0 BottleneckBlock-1 [[1, 64, 64, 64]] [1, 256, 64, 64] 0 Conv2D-6 [[1, 256, 64, 64]] [1, 256, 64, 64] 65,536 BatchNorm-6 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-6 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 Conv2D-7 [[1, 256, 64, 64]] [1, 256, 64, 64] 9,216 BatchNorm-7 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-7 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 Conv2D-8 [[1, 256, 64, 64]] [1, 256, 64, 64] 65,536 BatchNorm-8 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-8 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 BottleneckBlock-2 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 Conv2D-9 [[1, 256, 64, 64]] [1, 256, 64, 64] 65,536 BatchNorm-9 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-9 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 Conv2D-10 [[1, 256, 64, 64]] [1, 256, 64, 64] 9,216 BatchNorm-10 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-10 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 Conv2D-11 [[1, 256, 64, 64]] [1, 256, 64, 64] 65,536 BatchNorm-11 [[1, 256, 64, 64]] [1, 256, 64, 64] 1,024 ConvBNLayer-11 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 BottleneckBlock-3 [[1, 256, 64, 64]] [1, 256, 64, 64] 0 Conv2D-12 [[1, 256, 64, 64]] [1, 512, 64, 64] 131,072 BatchNorm-12 [[1, 512, 64, 64]] [1, 512, 64, 64] 2,048 ConvBNLayer-12 [[1, 256, 64, 64]] [1, 512, 64, 64] 0 Conv2D-13 [[1, 512, 64, 64]] [1, 512, 32, 32] 36,864 BatchNorm-13 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-13 [[1, 512, 64, 64]] [1, 512, 32, 32] 0 Conv2D-14 [[1, 512, 32, 32]] [1, 512, 32, 32] 262,144 BatchNorm-14 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-14 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-15 [[1, 256, 64, 64]] [1, 512, 32, 32] 131,072 BatchNorm-15 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-15 [[1, 256, 64, 64]] [1, 512, 32, 32] 0 BottleneckBlock-4 [[1, 256, 64, 64]] [1, 512, 32, 32] 0 Conv2D-16 [[1, 512, 32, 32]] [1, 512, 32, 32] 262,144 BatchNorm-16 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-16 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-17 [[1, 512, 32, 32]] [1, 512, 32, 32] 36,864 BatchNorm-17 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-17 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-18 [[1, 512, 32, 32]] [1, 512, 32, 32] 262,144 BatchNorm-18 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-18 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 BottleneckBlock-5 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-19 [[1, 512, 32, 32]] [1, 512, 32, 32] 262,144 BatchNorm-19 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-19 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-20 [[1, 512, 32, 32]] [1, 512, 32, 32] 36,864 BatchNorm-20 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-20 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-21 [[1, 512, 32, 32]] [1, 512, 32, 32] 262,144 BatchNorm-21 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-21 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 BottleneckBlock-6 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-22 [[1, 512, 32, 32]] [1, 512, 32, 32] 262,144 BatchNorm-22 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-22 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-23 [[1, 512, 32, 32]] [1, 512, 32, 32] 36,864 BatchNorm-23 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-23 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-24 [[1, 512, 32, 32]] [1, 512, 32, 32] 262,144 BatchNorm-24 [[1, 512, 32, 32]] [1, 512, 32, 32] 2,048 ConvBNLayer-24 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 BottleneckBlock-7 [[1, 512, 32, 32]] [1, 512, 32, 32] 0 Conv2D-25 [[1, 512, 32, 32]] [1, 1024, 32, 32] 524,288 BatchNorm-25 [[1, 1024, 32, 32]] [1, 1024, 32, 32] 4,096 ConvBNLayer-25 [[1, 512, 32, 32]] [1, 1024, 32, 32] 0 Conv2D-26 [[1, 1024, 32, 32]] [1, 1024, 16, 16] 147,456 BatchNorm-26 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-26 [[1, 1024, 32, 32]] [1, 1024, 16, 16] 0 Conv2D-27 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-27 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-27 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-28 [[1, 512, 32, 32]] [1, 1024, 16, 16] 524,288 BatchNorm-28 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-28 [[1, 512, 32, 32]] [1, 1024, 16, 16] 0 BottleneckBlock-8 [[1, 512, 32, 32]] [1, 1024, 16, 16] 0 Conv2D-29 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-29 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-29 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-30 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-30 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-30 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-31 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-31 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-31 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-9 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-32 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-32 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-32 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-33 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-33 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-33 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-34 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-34 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-34 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-10 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-35 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-35 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-35 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-36 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-36 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-36 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-37 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-37 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-37 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-11 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-38 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-38 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-38 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-39 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-39 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-39 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-40 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-40 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-40 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-12 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-41 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-41 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-41 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-42 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-42 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-42 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-43 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-43 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-43 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-13 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-44 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-44 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-44 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-45 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-45 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-45 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-46 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-46 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-46 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-14 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-47 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-47 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-47 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-48 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-48 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-48 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-49 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-49 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-49 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-15 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-50 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-50 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-50 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-51 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-51 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-51 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-52 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-52 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-52 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-16 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-53 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-53 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-53 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-54 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-54 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-54 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-55 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-55 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-55 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-17 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-56 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-56 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-56 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-57 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-57 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-57 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-58 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-58 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-58 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-18 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-59 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-59 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-59 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-60 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-60 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-60 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-61 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-61 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-61 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-19 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-62 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-62 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-62 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-63 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-63 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-63 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-64 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-64 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-64 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-20 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-65 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-65 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-65 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-66 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-66 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-66 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-67 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-67 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-67 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-21 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-68 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-68 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-68 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-69 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-69 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-69 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-70 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-70 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-70 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-22 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-71 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-71 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-71 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-72 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-72 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-72 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-73 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-73 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-73 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-23 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-74 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-74 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-74 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-75 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-75 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-75 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-76 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-76 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-76 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-24 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-77 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-77 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-77 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-78 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-78 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-78 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-79 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-79 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-79 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-25 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-80 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-80 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-80 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-81 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-81 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-81 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-82 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-82 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-82 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-26 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-83 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-83 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-83 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-84 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-84 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-84 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-85 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-85 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-85 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-27 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-86 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-86 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-86 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-87 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-87 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-87 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-88 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-88 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-88 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-28 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-89 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-89 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-89 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-90 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-90 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-90 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-91 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-91 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-91 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-29 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-92 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-92 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-92 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-93 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 147,456 BatchNorm-93 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-93 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-94 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 1,048,576 BatchNorm-94 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 4,096 ConvBNLayer-94 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 BottleneckBlock-30 [[1, 1024, 16, 16]] [1, 1024, 16, 16] 0 Conv2D-95 [[1, 1024, 16, 16]] [1, 2048, 16, 16] 2,097,152 BatchNorm-95 [[1, 2048, 16, 16]] [1, 2048, 16, 16] 8,192 ConvBNLayer-95 [[1, 1024, 16, 16]] [1, 2048, 16, 16] 0 Conv2D-96 [[1, 2048, 16, 16]] [1, 2048, 8, 8] 589,824 BatchNorm-96 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 8,192 ConvBNLayer-96 [[1, 2048, 16, 16]] [1, 2048, 8, 8] 0 Conv2D-97 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 4,194,304 BatchNorm-97 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 8,192 ConvBNLayer-97 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 0 Conv2D-98 [[1, 1024, 16, 16]] [1, 2048, 8, 8] 2,097,152 BatchNorm-98 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 8,192 ConvBNLayer-98 [[1, 1024, 16, 16]] [1, 2048, 8, 8] 0 BottleneckBlock-31 [[1, 1024, 16, 16]] [1, 2048, 8, 8] 0 Conv2D-99 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 4,194,304 BatchNorm-99 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 8,192 ConvBNLayer-99 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 0 Conv2D-100 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 589,824 BatchNorm-100 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 8,192 ConvBNLayer-100 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 0 Conv2D-101 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 4,194,304 BatchNorm-101 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 8,192 ConvBNLayer-101 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 0 BottleneckBlock-32 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 0 Conv2D-102 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 4,194,304 BatchNorm-102 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 8,192 ConvBNLayer-102 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 0 Conv2D-103 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 589,824 BatchNorm-103 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 8,192 ConvBNLayer-103 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 0 Conv2D-104 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 4,194,304 BatchNorm-104 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 8,192 ConvBNLayer-104 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 0 BottleneckBlock-33 [[1, 2048, 8, 8]] [1, 2048, 8, 8] 0 AdaptiveAvgPool2D-1 [[1, 2048, 8, 8]] [1, 2048, 1, 1] 0 Linear-1 [[1, 2048]] [1, 1000] 2,049,000 ResNeXt-2 [[1, 3, 256, 256]] [1, 1000] 0 Dropout-1 [[1, 1000]] [1, 1000] 0 Linear-2 [[1, 1000]] [1, 2] 2,002 =============================================================================== Total params: 83,660,154 Trainable params: 83,254,394 Non-trainable params: 405,760 ------------------------------------------------------------------------------- Input size (MB): 0.75 Forward/backward pass size (MB): 1024.04 Params size (MB): 319.14 Estimated Total Size (MB): 1343.93 -------------------------------------------------------------------------------
{'total_params': 83660154, 'trainable_params': 83254394}
|