src.gridmind.feature_construction.cnn_feature_extractor

Classes

ResNetFeatureExtractor

Module Contents

class src.gridmind.feature_construction.cnn_feature_extractor.ResNetFeatureExtractor(output_dim=512)[source]

Bases: torch.nn.Module

features[source]
pool[source]
flatten[source]
output_dim = 512[source]
forward(x)[source]