I am using the resnet3d18 4 available in the vision module:
    model = models.video.r3d_18(pretrained=True, progress=False)
    num_features = model.fc.in_features
    model.fc = nn.Linear(num_features, num_classes)

Since the model is being downloaded I am not sure how to label the layers and display the feature maps. Specifically I am trying to visualize the feature map of the first layer.
Can you all please help me out?
Thanks!