grab_layers_sequential

lightning_pose.models.backbones.torchvision.grab_layers_sequential(model, last_layer_ind: int) → Sequential

Package selected number of layers into a torch.nn.Sequential object.

Parameters:

• model – original resnet or efficientnet model

• last_layer_ind – final layer to pass data through

Returns:

potentially reduced backbone model