l5kit.environment.models module

l5kit.environment.models.SimpleCNN_GN(num_input_channels: int, features_dim: int) → torch.nn.modules.module.Module

A simplified feature extractor with GroupNorm.

Parameters

• num_input_channels – the number of input channels in the input

• features_dim – the number of features to extract from input