Conv2dRelu¶
Img224x224Kernel7x7SeparatedDims¶
-
class
Img224x224Kernel7x7SeparatedDims
(in_channels, out_params, norm_class, p_dropout=0)[source]¶ Bases:
shapenet.networks.abstract_network.AbstractFeatureExtractor
-
static
_build_model
(in_channels, out_params, norm_class, p_dropout)[source]¶ Build the actual model structure
Parameters: Returns: ensembled model
Return type: torch.nn.Module
-
forward
(input_batch)¶ Feed batch through network
Parameters: input_batch ( torch.Tensor
) – batch to feed through networkReturns: exracted features Return type: torch.Tensor
-
static