The Global Average Pooling

[jianjiandandande]

SKNet/sknet.py

Line 45 in ef76d8d

# fea_s = self.gap(fea_U).squeeze_()

SKNet/sknet.py

Line 46 in ef76d8d

fea_s = fea_U.mean(-1).mean(-1)

Does the code use mean instead of GAP? Is the performance obvious?

[midofalasol]

@jianjiandandande Hello, I have the same question, do you have an answer?

[Rayn-Wu]

To avoid error when batchsize = 1