fastreid.layers.pooling.GeneralizedMeanPoolingP Class Reference

Inheritance diagram for fastreid.layers.pooling.GeneralizedMeanPoolingP:

Public Member Functions
	__init__ (self, norm=3, output_size=1, eps=1e-6)
Public Member Functions inherited from fastreid.layers.pooling.GeneralizedMeanPooling
	forward (self, x)
	__repr__ (self)

Public Attributes
	p
Public Attributes inherited from fastreid.layers.pooling.GeneralizedMeanPooling
	p
	output_size
	eps

Detailed Description

 Same, but norm is trainable

Definition at line 56 of file pooling.py.

Constructor & Destructor Documentation

__init__()

fastreid.layers.pooling.GeneralizedMeanPoolingP.__init__	(		self,
			norm = 3,
			output_size = 1,
			eps = 1e-6 )

Reimplemented from fastreid.layers.pooling.GeneralizedMeanPooling.

Definition at line 60 of file pooling.py.

    def __init__(self, norm=3, output_size=1, eps=1e-6):
        super(GeneralizedMeanPoolingP, self).__init__(norm, output_size, eps)
        self.p = nn.Parameter(torch.ones(1) * norm)
 
 

Member Data Documentation

p

fastreid.layers.pooling.GeneralizedMeanPoolingP.p

Definition at line 62 of file pooling.py.

---

The documentation for this class was generated from the following file:

• smreid/fastreid/layers/pooling.py