Spaces:
Building
Building
File size: 313 Bytes
4187c6f |
1 2 3 4 5 6 7 8 9 10 11 12 |
name: feature_extractor_resnet
backbone:
pretrained: true
frozen: true
output_dim: ${model.latent_dim} # Match Latent Dimension
input_dim: 3
encoder: resnet50
num_downsample: null
remove_stride_from_first_conv: false
decoder_norm: "nn.BatchNorm2d"
do_average_pooling: false
checkpointed: false |