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