babel120 chkpts

babel_120/SAME_DOMAIN/best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86589eb3c99838b0d5ea74d81946d751bac7b5be5fa9f60e8682dde3dc8914d4
+size 690281230

babel_120/SAME_DOMAIN/config.yaml

@@ -0,0 +1,164 @@
+encoder:
+  name: dualcond_protogcn
+  params:
+    num_person: 1
+    in_channels: 2
+    num_prototype: 400
+    model_dim: 384
+    graph_cfg:
+      layout: 'coco'
+      mode: 'random'
+      num_filter: 8
+      init_off: .04
+      init_std: .02
+    use_pe: True
+
+head:
+  name: textembeddualoutput
+  params:
+    num_classes: 120
+    in_channels: 384
+    mode: 'contrastive'
+    text_embed_dim: 1536
+
+encoder_text:
+  name: text_encoder
+  params:
+    wpath: 'clip-ViT-B-32'
+    loader: 'sentence_transformer'
+    split_coma: true
+    n_splits: 3
+
+language:
+  mode: 'json'
+  json_file: "encoder/dataset/data/babel_mv/babel_gpt_12_angles_template.json"
+  indexes_keyword_train: '6'
+  indexes_keyword_test: '6'
+
+train:
+  optimizer:
+    name: adam
+    lr: 0.001
+    lr_decay: 0.1
+    milestones:
+    - 30
+  dataloader:
+    batch_size: 32
+    num_workers: 1
+    pin_memory: true
+    prefetch_factor: 2
+    persistent_workers: false
+  settings:
+    num_epochs: 50
+    save_interval: 1
+test:
+  dataloader:
+    batch_size: 1
+    num_workers: 1
+    pin_memory: true
+    prefetch_factor: 2
+    persistent_workers: false
+
+trainset:
+  dataset:
+    split: xsub_train
+    params:
+      n_views: 2
+      label_map: ../data/babel_mv/babel.txt
+      # motionid_labels_path: ../data/babel_mv/motionid_to_labels.pkl
+      num_classes: 120
+    name: babel_mv
+    use_negatives: true
+    file: ../data/babel_mv/babel_120_joints_extra_preprocessed.pkl
+  preprocessing:
+    KeepIndexes:
+      indexes:
+      - 0
+      random: true
+      append_ori: true
+    UniformSample:
+      clip_len: 150
+    PoseDecode: 0
+    AddNoiseToKeypoints:
+      doMasking: false
+      addScores: true
+    Collect:
+      keys:
+      - keypoint
+      - label
+      - indexes
+      - orientation
+      meta_keys: []
+    ToTensor:
+      keys:
+      - keypoint
+      - orientation
+
+testset:
+  dataset:
+    split: xsub_val
+    params:
+      n_views: 1
+      label_map: ../data/babel_mv/babel.txt
+      # motionid_labels_path: ../data/babel_mv/motionid_to_labels.pkl
+      num_classes: 120
+    name: babel_mv
+    file: ../data/babel_mv/babel_120_joints_extra_preprocessed.pkl
+  preprocessing:
+    KeepIndexes:
+      indexes:
+      - 0
+      append_ori: true
+    Collect:
+      keys:
+      - keypoint
+      - label
+      - orientation
+      meta_keys: []
+    ToTensor:
+      keys:
+      - keypoint
+      - orientation
+
+
+testset_all:
+  dataset:
+    split: xsub_val
+    params:
+      n_views: 1
+      label_map: ../data/babel_mv/babel.txt
+      # motionid_labels_path: ../data/babel_mv/motionid_to_labels.pkl
+      num_classes: 120
+    name: babel_mv
+    file: ../data/babel_mv/babel_120_joints_extra_preprocessed.pkl
+  preprocessing:
+    KeepIndexes:
+      indexes:
+      - 0
+      - 1
+      - 2
+      - 3
+      - 4
+      - 5
+      - 6
+      - 7
+      - 8
+      - 9
+      - 10
+      - 11
+      append_ori: true
+    Collect:
+      keys:
+      - keypoint
+      - label
+      - orientation
+      meta_keys: []
+    ToTensor:
+      keys:
+      - keypoint
+      - orientation
+
+
+frozen_encoder: null
+n_gpus: 1
+n_views: 1