Conv2dRelu¶
Img224x224Kernel7x7SeparatedDims¶
-
class
Img224x224Kernel7x7SeparatedDims
(in_channels, out_params, norm_class, p_dropout=0)[source]¶ Bases:
shapenet.jit.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.jit.ScriptModule
-
forward
(input_batch)¶ Feed batch through network
Parameters: input_batch ( torch.Tensor
) – batch to feed through networkReturns: extracted features Return type: torch.Tensor
-
static