_BASE_: "Base-DensePose-RCNN-FPN.yaml"
MODEL:
  WEIGHTS: "https://dl.fbaipublicfiles.com/densepose/cse/densepose_rcnn_R_50_FPN_soft_s1x/250533982/model_final_2c4512.pkl"
  RESNETS:
    DEPTH: 50
  ROI_DENSEPOSE_HEAD:
    NAME: "DensePoseV1ConvXHead"
    CSE:
      EMBED_LOSS_NAME: "SoftEmbeddingLoss"
      EMBEDDING_DIST_GAUSS_SIGMA: 0.1
      GEODESIC_DIST_GAUSS_SIGMA: 0.1
      EMBEDDERS:
        "chimp_5029":
          TYPE: vertex_feature
          NUM_VERTICES: 5029
          FEATURE_DIM: 256
          FEATURES_TRAINABLE: False
          IS_TRAINABLE: True
          INIT_FILE: "https://dl.fbaipublicfiles.com/densepose/data/cse/lbo/phi_chimp_5029_256.pkl"
DATASETS:
  TRAIN:
    - "densepose_chimps_cse_train"
  TEST:
    - "densepose_chimps_cse_val"
  CLASS_TO_MESH_NAME_MAPPING:
    "0": "chimp_5029"
SOLVER:
  MAX_ITER: 4000
  STEPS: (3000, 3500)