Upload 18 files

kit-amass-rot/1cp6dwpa/.hydra/config.yaml

@@ -0,0 +1,179 @@
+data:
+  transforms:
+    rots2rfeats:
+      name: SMPLVelP
+      _target_: temos.transforms.rots2rfeats.SMPLVelP
+      pose_rep: rot6d
+      path: ${path.deps}/transforms/rots2rfeats/smplvelp/${.pose_rep}/${data.dataname}
+      canonicalize: true
+      normalization: ${transforms.normalization}
+    rots2joints:
+      name: SMPLH
+      _target_: temos.transforms.rots2joints.SMPLH
+      jointstype: mmm
+      input_pose_rep: matrix
+      path: ${path.deps}/smplh
+      batch_size: ${machine.smpl_batch_size}
+      gender: neutral
+    joints2jfeats:
+      name: Rifke
+      _target_: temos.transforms.joints2jfeats.Rifke
+      path: ${path.deps}/transforms/joints2jfeats/rifke/${data.dataname}
+      normalization: ${transforms.normalization}
+      forward_filter: false
+    name: SMPLTransform
+    ename: smpl
+    _target_: temos.transforms.SMPLTransform
+    normalization: true
+  dataname: kit-amass-rot
+  _target_: temos.data.kit.KITDataModule
+  datapath: ${path.datasets}/kit
+  splitpath: ${path.datasets}/kit-splits
+  load_amass_data: true
+  load_with_rot: true
+  correspondance_path: ${path.datasets}/kitml_amass_path.json
+  amass_path: ${path.datasets}/AMASS
+  smplh_path: ${path.deps}/smplh
+  pick_one_text: true
+  batch_size: ${machine.batch_size}
+  num_workers: ${machine.num_workers}
+  framerate: 12.5
+  sampler: ${sampler}
+  tiny: false
+  progress_bar: true
+model:
+  textencoder:
+    name: distilbert_actor
+    _target_: temos.model.textencoder.distilbert_actor.DistilbertActorAgnosticEncoder
+    latent_dim: ${model.latent_dim}
+    vae: ${model.vae}
+    ff_size: ${model.ff_size}
+    num_layers: ${model.num_layers}
+    num_head: ${model.num_head}
+    droupout: ${model.droupout}
+    activation: ${model.activation}
+    finetune: false
+    modelpath: ${path.deps}/distilbert-base-uncased
+  motionencoder:
+    name: actor_encoder
+    _target_: temos.model.motionencoder.actor.ActorAgnosticEncoder
+    latent_dim: ${model.latent_dim}
+    vae: ${model.vae}
+    ff_size: ${model.ff_size}
+    num_layers: ${model.num_layers}
+    num_head: ${model.num_head}
+    droupout: ${model.droupout}
+    activation: ${model.activation}
+  motiondecoder:
+    name: actor_decoder
+    _target_: temos.model.motiondecoder.actor.ActorAgnosticDecoder
+    latent_dim: ${model.latent_dim}
+    ff_size: ${model.ff_size}
+    num_layers: ${model.num_layers}
+    num_head: ${model.num_head}
+    droupout: ${model.droupout}
+    activation: ${model.activation}
+  losses:
+    _target_: temos.model.losses.TemosComputeLosses
+    mode: ${transforms.ename}
+    lmd_rfeats_recons: 1.0
+    lmd_jfeats_recons: 1.0
+    lmd_latent: 1.0e-05
+    lmd_kl: 1.0e-05
+    loss_on_both: true
+    force_loss_on_jfeats: false
+    ablation_no_kl_combine: false
+    ablation_no_kl_gaussian: false
+    ablation_no_motionencoder: false
+    recons_text2rfeats: ${.lmd_rfeats_recons}
+    recons_text2rfeats_func: ${model.func_recons}
+    recons_text2jfeats: ${.lmd_jfeats_recons}
+    recons_text2jfeats_func: ${model.func_recons}
+    recons_rfeats2rfeats: ${.lmd_rfeats_recons}
+    recons_rfeats2rfeats_func: ${model.func_recons}
+    recons_jfeats2jfeats: ${.lmd_jfeats_recons}
+    recons_jfeats2jfeats_func: ${model.func_recons}
+    latent_manifold: ${.lmd_latent}
+    latent_manifold_func: ${model.func_latent}
+    kl_text: ${.lmd_kl}
+    kl_text_func: ${model.func_kl}
+    kl_motion: ${.lmd_kl}
+    kl_motion_func: ${model.func_kl}
+    kl_text2motion: ${.lmd_kl}
+    kl_text2motion_func: ${model.func_kl}
+    kl_motion2text: ${.lmd_kl}
+    kl_motion2text_func: ${model.func_kl}
+  optim:
+    _target_: torch.optim.AdamW
+    lr: 0.0001
+  func_recons:
+    _target_: torch.nn.SmoothL1Loss
+    reduction: mean
+  func_latent:
+    _target_: torch.nn.SmoothL1Loss
+    reduction: mean
+  func_kl:
+    _target_: temos.model.losses.KLLoss
+  modelname: temos
+  _target_: temos.model.temos.TEMOS
+  latent_dim: 256
+  vae: true
+  transforms: ${transforms}
+  ff_size: 1024
+  num_layers: 6
+  num_head: ${model.num_layers}
+  droupout: 0.1
+  activation: gelu
+machine:
+  name: server
+  batch_size: 32
+  smpl_batch_size: 256
+  num_workers: 8
+trainer:
+  auto_select_gpus: true
+  benchmark: false
+  max_epochs: 1001
+  accelerator: gpu
+  devices: 1
+  log_every_n_steps: 1
+  deterministic: false
+  detect_anomaly: false
+  enable_progress_bar: true
+sampler:
+  _target_: temos.data.sampling.FrameSampler
+  request_frames: null
+  sampling: conseq
+  sampling_step: 1
+  max_len: 500
+  min_len: 10
+logger:
+  logger_name: none
+  version: ${run_id}
+  project: null
+callback:
+  last_ckpt:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    filename: latest-{epoch}
+    every_n_epochs: 1
+    save_top_k: 1
+    save_last: true
+  latest_ckpt:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    filename: latest-{epoch}
+    monitor: step
+    mode: max
+    every_n_epochs: 500
+    save_top_k: -1
+    save_last: false
+  progress:
+    _target_: temos.callback.ProgressLogger
+path:
+  deps: ${code_path:./deps}
+  datasets: ${code_path:./datasets}
+  code_dir: ${code_path:}
+  working_dir: ${working_path:""}
+experiment: baseline
+seed: 1234
+logger_level: INFO
+run_id: ${generate_id:}
+transforms: ${data.transforms}

kit-amass-rot/1cp6dwpa/.hydra/hydra.yaml

@@ -0,0 +1,207 @@
+hydra:
+  run:
+    dir: outputs/${data.dataname}/${experiment}/${run_id}
+  sweep:
+    dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
+    subdir: ${hydra.job.num}
+  launcher:
+    _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
+  sweeper:
+    _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
+    max_batch_size: null
+    params: null
+  help:
+    app_name: ${hydra.job.name}
+    header: '${hydra.help.app_name} is powered by Hydra.
+
+      '
+    footer: 'Powered by Hydra (https://hydra.cc)
+
+      Use --hydra-help to view Hydra specific help
+
+      '
+    template: '${hydra.help.header}
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (group=option)
+
+
+      $APP_CONFIG_GROUPS
+
+
+      == Config ==
+
+      Override anything in the config (foo.bar=value)
+
+
+      $CONFIG
+
+
+      ${hydra.help.footer}
+
+      '
+  hydra_help:
+    template: 'Hydra (${hydra.runtime.version})
+
+      See https://hydra.cc for more info.
+
+
+      == Flags ==
+
+      $FLAGS_HELP
+
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (For example, append hydra/job_logging=disabled
+      to command line)
+
+
+      $HYDRA_CONFIG_GROUPS
+
+
+      Use ''--cfg hydra'' to Show the Hydra config.
+
+      '
+    hydra_help: ???
+  hydra_logging:
+    version: 1
+    formatters:
+      colorlog:
+        (): colorlog.ColoredFormatter
+        format: '[%(cyan)s%(asctime)s%(reset)s][%(purple)sHYDRA%(reset)s] %(message)s'
+        datefmt: '%d/%m/%y %H:%M:%S'
+    handlers:
+      console:
+        class: rich.logging.RichHandler
+        formatter: colorlog
+    root:
+      level: INFO
+      handlers:
+      - console
+    disable_existing_loggers: false
+  job_logging:
+    version: 1
+    filters:
+      onlyimportant:
+        (): temos.tools.logging.LevelsFilter
+        levels:
+        - CRITICAL
+        - ERROR
+        - WARNING
+      noimportant:
+        (): temos.tools.logging.LevelsFilter
+        levels:
+        - INFO
+        - DEBUG
+        - NOTSET
+    formatters:
+      verysimple:
+        format: '%(message)s'
+      simple:
+        format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
+        datefmt: '%d/%m/%y %H:%M:%S'
+      colorlog:
+        (): colorlog.ColoredFormatter
+        format: '[%(cyan)s%(asctime)s%(reset)s][%(blue)s%(name)s%(reset)s][%(log_color)s%(levelname)s%(reset)s]
+          - %(message)s'
+        datefmt: '%d/%m/%y %H:%M:%S'
+      log_colors:
+        DEBUG: purple
+        INFO: green
+        WARNING: yellow
+        ERROR: red
+        CRITICAL: red
+    handlers:
+      console:
+        class: rich.logging.RichHandler
+        formatter: verysimple
+        rich_tracebacks: true
+      file_out:
+        class: logging.FileHandler
+        formatter: simple
+        filename: logs.out
+        filters:
+        - noimportant
+      file_err:
+        class: logging.FileHandler
+        formatter: simple
+        filename: logs.err
+        filters:
+        - onlyimportant
+    root:
+      level: ${logger_level}
+      handlers:
+      - console
+      - file_out
+      - file_err
+    disable_existing_loggers: false
+  env: {}
+  mode: RUN
+  searchpath: []
+  callbacks: {}
+  output_subdir: .hydra
+  overrides:
+    hydra:
+    - hydra.mode=RUN
+    task:
+    - data=kit-amass-rot
+  job:
+    name: train
+    chdir: true
+    override_dirname: data=kit-amass-rot
+    id: ???
+    num: ???
+    config_name: train
+    env_set: {}
+    env_copy: []
+    config:
+      override_dirname:
+        kv_sep: '='
+        item_sep: ','
+        exclude_keys: []
+  runtime:
+    version: 1.2.0
+    version_base: '1.2'
+    cwd: /home/mathis/TEMOS
+    config_sources:
+    - path: hydra.conf
+      schema: pkg
+      provider: hydra
+    - path: /home/mathis/TEMOS/configs
+      schema: file
+      provider: main
+    - path: hydra_plugins.hydra_colorlog.conf
+      schema: pkg
+      provider: hydra-colorlog
+    - path: ''
+      schema: structured
+      provider: schema
+    output_dir: /home/mathis/TEMOS/outputs/kit-amass-rot/baseline/1cp6dwpa
+    choices:
+      callback: base
+      logger: none
+      sampler: variable_conseq
+      trainer: gpu
+      machine: server
+      model: temos
+      model/optim: adamw
+      model/losses: temos
+      model/motiondecoder: actor
+      model/motionencoder: actor
+      model/textencoder: distilbert_actor
+      data: kit-amass-rot
+      transforms/joints2jfeats@data.transforms.joints2jfeats: rifke
+      transforms/rots2joints@data.transforms.rots2joints: smplh
+      transforms/rots2rfeats@data.transforms.rots2rfeats: smplvelp
+      hydra/env: default
+      hydra/callbacks: null
+      hydra/job_logging: rich
+      hydra/hydra_logging: rich
+      hydra/hydra_help: default
+      hydra/help: default
+      hydra/sweeper: basic
+      hydra/launcher: basic
+      hydra/output: default
+  verbose: false

kit-amass-rot/1cp6dwpa/.hydra/overrides.yaml

@@ -0,0 +1 @@
+- data=kit-amass-rot

kit-amass-rot/1cp6dwpa/checkpoints/last.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:703fad94ed0ec9ebfdd2a38fd4dff6e39112d466480797ac61afd80fb3b31c85
+size 516258667

kit-amass-xyz/5xp9647f/.hydra/config.yaml

@@ -0,0 +1,203 @@
+data:
+  transforms_xyz:
+    joints2jfeats:
+      name: Rifke
+      _target_: temos.transforms.joints2jfeats.Rifke
+      path: ${path.deps}/transforms/joints2jfeats/rifke/${data.dataname}
+      normalization: ${transforms.normalization}
+      forward_filter: false
+    name: XYZTransform
+    ename: xyz
+    _target_: temos.transforms.xyz.XYZTransform
+    normalization: true
+  transforms_smpl:
+    rots2rfeats:
+      name: SMPLVelP
+      _target_: temos.transforms.rots2rfeats.SMPLVelP
+      pose_rep: rot6d
+      path: ${path.deps}/transforms/rots2rfeats/smplvelp/${.pose_rep}/${data.dataname}
+      canonicalize: true
+      normalization: false
+    rots2joints:
+      name: SMPLH
+      _target_: temos.transforms.rots2joints.SMPLH
+      jointstype: mmm
+      input_pose_rep: matrix
+      path: ${path.deps}/smplh
+      batch_size: ${machine.smpl_batch_size}
+      gender: neutral
+    joints2jfeats:
+      name: Rifke
+      _target_: temos.transforms.joints2jfeats.Rifke
+      path: ${path.deps}/transforms/joints2jfeats/rifke/${data.dataname}
+      normalization: false
+      forward_filter: false
+    name: SMPLTransform
+    ename: smpl
+    _target_: temos.transforms.SMPLTransform
+    normalization: false
+  dataname: kit-amass-xyz
+  _target_: temos.data.kit.KITDataModule
+  datapath: ${path.datasets}/kit
+  splitpath: ${path.datasets}/kit-splits
+  load_amass_data: true
+  load_with_rot: false
+  correspondance_path: ${path.datasets}/kitml_amass_path.json
+  amass_path: ${path.datasets}/AMASS
+  smplh_path: ${path.deps}/smplh
+  load_actions: false
+  load_canonicalized_text: false
+  pick_one_text: true
+  batch_size: ${machine.batch_size}
+  num_workers: ${machine.num_workers}
+  framerate: 12.5
+  sampler: ${sampler}
+  transforms:
+    joints2jfeats:
+      name: Rifke
+      _target_: temos.transforms.joints2jfeats.Rifke
+      path: ${path.deps}/transforms/joints2jfeats/rifke/${data.dataname}
+      normalization: ${transforms.normalization}
+      forward_filter: false
+    name: XYZTransform
+    ename: xyz
+    _target_: temos.transforms.xyz.XYZTransform
+    normalization: true
+  tiny: false
+  progress_bar: true
+model:
+  textencoder:
+    name: distilbert_actor
+    _target_: temos.model.textencoder.distilbert_actor.DistilbertActorAgnosticEncoder
+    latent_dim: ${model.latent_dim}
+    vae: ${model.vae}
+    ff_size: ${model.ff_size}
+    num_layers: ${model.num_layers}
+    num_head: ${model.num_head}
+    droupout: ${model.droupout}
+    activation: ${model.activation}
+    finetune: false
+    modelpath: ${path.deps}/distilbert-base-uncased
+  motionencoder:
+    name: actor_encoder
+    _target_: temos.model.motionencoder.actor.ActorAgnosticEncoder
+    latent_dim: ${model.latent_dim}
+    vae: ${model.vae}
+    ff_size: ${model.ff_size}
+    num_layers: ${model.num_layers}
+    num_head: ${model.num_head}
+    droupout: ${model.droupout}
+    activation: ${model.activation}
+  motiondecoder:
+    name: actor_decoder
+    _target_: temos.model.motiondecoder.actor.ActorAgnosticDecoder
+    latent_dim: ${model.latent_dim}
+    ff_size: ${model.ff_size}
+    num_layers: ${model.num_layers}
+    num_head: ${model.num_head}
+    droupout: ${model.droupout}
+    activation: ${model.activation}
+  losses:
+    _target_: temos.model.losses.TemosComputeLosses
+    mode: ${transforms.ename}
+    lmd_rfeats_recons: 1.0
+    lmd_jfeats_recons: 1.0
+    lmd_latent: 1.0e-05
+    lmd_kl: 1.0e-05
+    loss_on_both: true
+    force_loss_on_jfeats: false
+    ablation_no_kl_combine: false
+    ablation_no_kl_gaussian: false
+    ablation_no_motionencoder: false
+    recons_text2rfeats: ${.lmd_rfeats_recons}
+    recons_text2rfeats_func: ${model.func_recons}
+    recons_text2jfeats: ${.lmd_jfeats_recons}
+    recons_text2jfeats_func: ${model.func_recons}
+    recons_rfeats2rfeats: ${.lmd_rfeats_recons}
+    recons_rfeats2rfeats_func: ${model.func_recons}
+    recons_jfeats2jfeats: ${.lmd_jfeats_recons}
+    recons_jfeats2jfeats_func: ${model.func_recons}
+    latent_manifold: ${.lmd_latent}
+    latent_manifold_func: ${model.func_latent}
+    kl_text: ${.lmd_kl}
+    kl_text_func: ${model.func_kl}
+    kl_motion: ${.lmd_kl}
+    kl_motion_func: ${model.func_kl}
+    kl_text2motion: ${.lmd_kl}
+    kl_text2motion_func: ${model.func_kl}
+    kl_motion2text: ${.lmd_kl}
+    kl_motion2text_func: ${model.func_kl}
+  optim:
+    _target_: torch.optim.AdamW
+    lr: 0.0001
+  func_recons:
+    _target_: torch.nn.SmoothL1Loss
+    reduction: mean
+  func_latent:
+    _target_: torch.nn.SmoothL1Loss
+    reduction: mean
+  func_kl:
+    _target_: temos.model.losses.KLLoss
+  modelname: temos
+  _target_: temos.model.temos.TEMOS
+  latent_dim: 256
+  vae: true
+  transforms: ${transforms}
+  ff_size: 1024
+  num_layers: 6
+  num_head: ${model.num_layers}
+  droupout: 0.1
+  activation: gelu
+machine:
+  name: server
+  batch_size: 32
+  smpl_batch_size: 256
+  num_workers: 8
+trainer:
+  auto_select_gpus: true
+  benchmark: false
+  max_epochs: 1001
+  accelerator: gpu
+  devices: 1
+  log_every_n_steps: 1
+  deterministic: false
+  detect_anomaly: false
+  enable_progress_bar: true
+sampler:
+  _target_: temos.data.sampling.FrameSampler
+  request_frames: null
+  sampling: conseq
+  sampling_step: 1
+  max_len: 500
+  min_len: 10
+logger:
+  logger_name: none
+  version: ${run_id}
+  project: null
+callback:
+  last_ckpt:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    filename: latest-{epoch}
+    every_n_epochs: 1
+    save_top_k: 1
+    save_last: true
+  latest_ckpt:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    filename: latest-{epoch}
+    monitor: step
+    mode: max
+    every_n_epochs: 500
+    save_top_k: -1
+    save_last: false
+  progress:
+    _target_: temos.callback.ProgressLogger
+path:
+  deps: ${code_path:./deps}
+  datasets: ${code_path:./datasets}
+  code_dir: ${code_path:}
+  working_dir: ${working_path:""}
+experiment: baseline
+seed: 1234
+logger_level: INFO
+run_id: ${generate_id:}
+transforms: ${data.transforms}

kit-amass-xyz/5xp9647f/.hydra/hydra.yaml

@@ -0,0 +1,208 @@
+hydra:
+  run:
+    dir: outputs/${data.dataname}/${experiment}/${run_id}
+  sweep:
+    dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
+    subdir: ${hydra.job.num}
+  launcher:
+    _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
+  sweeper:
+    _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
+    max_batch_size: null
+    params: null
+  help:
+    app_name: ${hydra.job.name}
+    header: '${hydra.help.app_name} is powered by Hydra.
+
+      '
+    footer: 'Powered by Hydra (https://hydra.cc)
+
+      Use --hydra-help to view Hydra specific help
+
+      '
+    template: '${hydra.help.header}
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (group=option)
+
+
+      $APP_CONFIG_GROUPS
+
+
+      == Config ==
+
+      Override anything in the config (foo.bar=value)
+
+
+      $CONFIG
+
+
+      ${hydra.help.footer}
+
+      '
+  hydra_help:
+    template: 'Hydra (${hydra.runtime.version})
+
+      See https://hydra.cc for more info.
+
+
+      == Flags ==
+
+      $FLAGS_HELP
+
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (For example, append hydra/job_logging=disabled
+      to command line)
+
+
+      $HYDRA_CONFIG_GROUPS
+
+
+      Use ''--cfg hydra'' to Show the Hydra config.
+
+      '
+    hydra_help: ???
+  hydra_logging:
+    version: 1
+    formatters:
+      colorlog:
+        (): colorlog.ColoredFormatter
+        format: '[%(cyan)s%(asctime)s%(reset)s][%(purple)sHYDRA%(reset)s] %(message)s'
+        datefmt: '%d/%m/%y %H:%M:%S'
+    handlers:
+      console:
+        class: rich.logging.RichHandler
+        formatter: colorlog
+    root:
+      level: INFO
+      handlers:
+      - console
+    disable_existing_loggers: false
+  job_logging:
+    version: 1
+    filters:
+      onlyimportant:
+        (): temos.tools.logging.LevelsFilter
+        levels:
+        - CRITICAL
+        - ERROR
+        - WARNING
+      noimportant:
+        (): temos.tools.logging.LevelsFilter
+        levels:
+        - INFO
+        - DEBUG
+        - NOTSET
+    formatters:
+      verysimple:
+        format: '%(message)s'
+      simple:
+        format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
+        datefmt: '%d/%m/%y %H:%M:%S'
+      colorlog:
+        (): colorlog.ColoredFormatter
+        format: '[%(cyan)s%(asctime)s%(reset)s][%(blue)s%(name)s%(reset)s][%(log_color)s%(levelname)s%(reset)s]
+          - %(message)s'
+        datefmt: '%d/%m/%y %H:%M:%S'
+      log_colors:
+        DEBUG: purple
+        INFO: green
+        WARNING: yellow
+        ERROR: red
+        CRITICAL: red
+    handlers:
+      console:
+        class: rich.logging.RichHandler
+        formatter: verysimple
+        rich_tracebacks: true
+      file_out:
+        class: logging.FileHandler
+        formatter: simple
+        filename: logs.out
+        filters:
+        - noimportant
+      file_err:
+        class: logging.FileHandler
+        formatter: simple
+        filename: logs.err
+        filters:
+        - onlyimportant
+    root:
+      level: ${logger_level}
+      handlers:
+      - console
+      - file_out
+      - file_err
+    disable_existing_loggers: false
+  env: {}
+  mode: RUN
+  searchpath: []
+  callbacks: {}
+  output_subdir: .hydra
+  overrides:
+    hydra:
+    - hydra.mode=RUN
+    task:
+    - data=kit-amass-xyz
+  job:
+    name: train
+    chdir: true
+    override_dirname: data=kit-amass-xyz
+    id: ???
+    num: ???
+    config_name: train
+    env_set: {}
+    env_copy: []
+    config:
+      override_dirname:
+        kv_sep: '='
+        item_sep: ','
+        exclude_keys: []
+  runtime:
+    version: 1.2.0
+    version_base: '1.2'
+    cwd: /home/mathis/TEMOS
+    config_sources:
+    - path: hydra.conf
+      schema: pkg
+      provider: hydra
+    - path: /home/mathis/TEMOS/configs
+      schema: file
+      provider: main
+    - path: hydra_plugins.hydra_colorlog.conf
+      schema: pkg
+      provider: hydra-colorlog
+    - path: ''
+      schema: structured
+      provider: schema
+    output_dir: /home/mathis/TEMOS/outputs/kit-amass-xyz/baseline/5xp9647f
+    choices:
+      callback: base
+      logger: none
+      sampler: variable_conseq
+      trainer: gpu
+      machine: server
+      model: temos
+      model/optim: adamw
+      model/losses: temos
+      model/motiondecoder: actor
+      model/motionencoder: actor
+      model/textencoder: distilbert_actor
+      data: kit-amass-xyz
+      transforms/joints2jfeats@data.transforms_smpl.joints2jfeats: rifke
+      transforms/rots2joints@data.transforms_smpl.rots2joints: smplh
+      transforms/rots2rfeats@data.transforms_smpl.rots2rfeats: smplvelp
+      transforms/joints2jfeats@data.transforms_xyz.joints2jfeats: rifke
+      hydra/env: default
+      hydra/callbacks: null
+      hydra/job_logging: rich
+      hydra/hydra_logging: rich
+      hydra/hydra_help: default
+      hydra/help: default
+      hydra/sweeper: basic
+      hydra/launcher: basic
+      hydra/output: default
+  verbose: false

kit-amass-xyz/5xp9647f/.hydra/overrides.yaml

@@ -0,0 +1 @@
+- data=kit-amass-xyz

kit-amass-xyz/5xp9647f/checkpoints/last.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:976aa973604eb4090fe8a02158b8eaae91f8dd8e9adfe140f07941e9e5bf6277
+size 515819115

kit-mmm-xyz/3l49g7hv/.hydra/config.yaml

@@ -0,0 +1,161 @@
+data:
+  transforms:
+    joints2jfeats:
+      name: Rifke
+      _target_: temos.transforms.joints2jfeats.Rifke
+      path: ${path.deps}/transforms/joints2jfeats/rifke/${data.dataname}
+      normalization: ${transforms.normalization}
+      forward_filter: false
+    name: XYZTransform
+    ename: xyz
+    _target_: temos.transforms.xyz.XYZTransform
+    normalization: true
+  dataname: kit-mmm-xyz
+  _target_: temos.data.kit.KITDataModule
+  datapath: ${path.datasets}/kit
+  splitpath: ${path.datasets}/kit-splits
+  load_amass_data: false
+  load_with_rot: false
+  pick_one_text: true
+  batch_size: ${machine.batch_size}
+  num_workers: ${machine.num_workers}
+  framerate: 12.5
+  sampler: ${sampler}
+  tiny: false
+  progress_bar: true
+model:
+  textencoder:
+    name: distilbert_actor
+    _target_: temos.model.textencoder.distilbert_actor.DistilbertActorAgnosticEncoder
+    latent_dim: ${model.latent_dim}
+    vae: ${model.vae}
+    ff_size: ${model.ff_size}
+    num_layers: ${model.num_layers}
+    num_head: ${model.num_head}
+    droupout: ${model.droupout}
+    activation: ${model.activation}
+    finetune: false
+    modelpath: ${path.deps}/distilbert-base-uncased
+  motionencoder:
+    name: actor_encoder
+    _target_: temos.model.motionencoder.actor.ActorAgnosticEncoder
+    latent_dim: ${model.latent_dim}
+    vae: ${model.vae}
+    ff_size: ${model.ff_size}
+    num_layers: ${model.num_layers}
+    num_head: ${model.num_head}
+    droupout: ${model.droupout}
+    activation: ${model.activation}
+  motiondecoder:
+    name: actor_decoder
+    _target_: temos.model.motiondecoder.actor.ActorAgnosticDecoder
+    latent_dim: ${model.latent_dim}
+    ff_size: ${model.ff_size}
+    num_layers: ${model.num_layers}
+    num_head: ${model.num_head}
+    droupout: ${model.droupout}
+    activation: ${model.activation}
+  losses:
+    _target_: temos.model.losses.TemosComputeLosses
+    mode: ${transforms.ename}
+    lmd_rfeats_recons: 1.0
+    lmd_jfeats_recons: 1.0
+    lmd_latent: 1.0e-05
+    lmd_kl: 1.0e-05
+    loss_on_both: true
+    force_loss_on_jfeats: false
+    ablation_no_kl_combine: false
+    ablation_no_kl_gaussian: false
+    ablation_no_motionencoder: false
+    recons_text2rfeats: ${.lmd_rfeats_recons}
+    recons_text2rfeats_func: ${model.func_recons}
+    recons_text2jfeats: ${.lmd_jfeats_recons}
+    recons_text2jfeats_func: ${model.func_recons}
+    recons_rfeats2rfeats: ${.lmd_rfeats_recons}
+    recons_rfeats2rfeats_func: ${model.func_recons}
+    recons_jfeats2jfeats: ${.lmd_jfeats_recons}
+    recons_jfeats2jfeats_func: ${model.func_recons}
+    latent_manifold: ${.lmd_latent}
+    latent_manifold_func: ${model.func_latent}
+    kl_text: ${.lmd_kl}
+    kl_text_func: ${model.func_kl}
+    kl_motion: ${.lmd_kl}
+    kl_motion_func: ${model.func_kl}
+    kl_text2motion: ${.lmd_kl}
+    kl_text2motion_func: ${model.func_kl}
+    kl_motion2text: ${.lmd_kl}
+    kl_motion2text_func: ${model.func_kl}
+  optim:
+    _target_: torch.optim.AdamW
+    lr: 0.0001
+  func_recons:
+    _target_: torch.nn.SmoothL1Loss
+    reduction: mean
+  func_latent:
+    _target_: torch.nn.SmoothL1Loss
+    reduction: mean
+  func_kl:
+    _target_: temos.model.losses.KLLoss
+  modelname: temos
+  _target_: temos.model.temos.TEMOS
+  latent_dim: 256
+  vae: true
+  transforms: ${transforms}
+  ff_size: 1024
+  num_layers: 6
+  num_head: ${model.num_layers}
+  droupout: 0.1
+  activation: gelu
+machine:
+  name: server
+  batch_size: 32
+  smpl_batch_size: 256
+  num_workers: 8
+trainer:
+  auto_select_gpus: true
+  benchmark: false
+  max_epochs: 1001
+  accelerator: gpu
+  devices: 1
+  log_every_n_steps: 1
+  deterministic: false
+  detect_anomaly: false
+  enable_progress_bar: true
+sampler:
+  _target_: temos.data.sampling.FrameSampler
+  request_frames: null
+  sampling: conseq
+  sampling_step: 1
+  max_len: 500
+  min_len: 10
+logger:
+  logger_name: none
+  version: ${run_id}
+  project: null
+callback:
+  last_ckpt:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    filename: latest-{epoch}
+    every_n_epochs: 1
+    save_top_k: 1
+    save_last: true
+  latest_ckpt:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    filename: latest-{epoch}
+    monitor: step
+    mode: max
+    every_n_epochs: 500
+    save_top_k: -1
+    save_last: false
+  progress:
+    _target_: temos.callback.ProgressLogger
+path:
+  deps: ${code_path:./deps}
+  datasets: ${code_path:./datasets}
+  code_dir: ${code_path:}
+  working_dir: ${working_path:""}
+experiment: baseline
+seed: 1234
+logger_level: INFO
+run_id: ${generate_id:}
+transforms: ${data.transforms}

kit-mmm-xyz/3l49g7hv/.hydra/hydra.yaml

@@ -0,0 +1,204 @@
+hydra:
+  run:
+    dir: outputs/${data.dataname}/${experiment}/${run_id}
+  sweep:
+    dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
+    subdir: ${hydra.job.num}
+  launcher:
+    _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
+  sweeper:
+    _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
+    max_batch_size: null
+    params: null
+  help:
+    app_name: ${hydra.job.name}
+    header: '${hydra.help.app_name} is powered by Hydra.
+
+      '
+    footer: 'Powered by Hydra (https://hydra.cc)
+
+      Use --hydra-help to view Hydra specific help
+
+      '
+    template: '${hydra.help.header}
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (group=option)
+
+
+      $APP_CONFIG_GROUPS
+
+
+      == Config ==
+
+      Override anything in the config (foo.bar=value)
+
+
+      $CONFIG
+
+
+      ${hydra.help.footer}
+
+      '
+  hydra_help:
+    template: 'Hydra (${hydra.runtime.version})
+
+      See https://hydra.cc for more info.
+
+
+      == Flags ==
+
+      $FLAGS_HELP
+
+
+      == Configuration groups ==
+
+      Compose your configuration from those groups (For example, append hydra/job_logging=disabled
+      to command line)
+
+
+      $HYDRA_CONFIG_GROUPS
+
+
+      Use ''--cfg hydra'' to Show the Hydra config.
+
+      '
+    hydra_help: ???
+  hydra_logging:
+    version: 1
+    formatters:
+      colorlog:
+        (): colorlog.ColoredFormatter
+        format: '[%(cyan)s%(asctime)s%(reset)s][%(purple)sHYDRA%(reset)s] %(message)s'
+        datefmt: '%d/%m/%y %H:%M:%S'
+    handlers:
+      console:
+        class: rich.logging.RichHandler
+        formatter: colorlog
+    root:
+      level: INFO
+      handlers:
+      - console
+    disable_existing_loggers: false
+  job_logging:
+    version: 1
+    filters:
+      onlyimportant:
+        (): temos.tools.logging.LevelsFilter
+        levels:
+        - CRITICAL
+        - ERROR
+        - WARNING
+      noimportant:
+        (): temos.tools.logging.LevelsFilter
+        levels:
+        - INFO
+        - DEBUG
+        - NOTSET
+    formatters:
+      verysimple:
+        format: '%(message)s'
+      simple:
+        format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
+        datefmt: '%d/%m/%y %H:%M:%S'
+      colorlog:
+        (): colorlog.ColoredFormatter
+        format: '[%(cyan)s%(asctime)s%(reset)s][%(blue)s%(name)s%(reset)s][%(log_color)s%(levelname)s%(reset)s]
+          - %(message)s'
+        datefmt: '%d/%m/%y %H:%M:%S'
+      log_colors:
+        DEBUG: purple
+        INFO: green
+        WARNING: yellow
+        ERROR: red
+        CRITICAL: red
+    handlers:
+      console:
+        class: rich.logging.RichHandler
+        formatter: verysimple
+        rich_tracebacks: true
+      file_out:
+        class: logging.FileHandler
+        formatter: simple
+        filename: logs.out
+        filters:
+        - noimportant
+      file_err:
+        class: logging.FileHandler
+        formatter: simple
+        filename: logs.err
+        filters:
+        - onlyimportant
+    root:
+      level: ${logger_level}
+      handlers:
+      - console
+      - file_out
+      - file_err
+    disable_existing_loggers: false
+  env: {}
+  mode: RUN
+  searchpath: []
+  callbacks: {}
+  output_subdir: .hydra
+  overrides:
+    hydra:
+    - hydra.mode=RUN
+    task: []
+  job:
+    name: train
+    chdir: true
+    override_dirname: ''
+    id: ???
+    num: ???
+    config_name: train
+    env_set: {}
+    env_copy: []
+    config:
+      override_dirname:
+        kv_sep: '='
+        item_sep: ','
+        exclude_keys: []
+  runtime:
+    version: 1.2.0
+    version_base: '1.2'
+    cwd: /home/mathis/TEMOS
+    config_sources:
+    - path: hydra.conf
+      schema: pkg
+      provider: hydra
+    - path: /home/mathis/TEMOS/configs
+      schema: file
+      provider: main
+    - path: hydra_plugins.hydra_colorlog.conf
+      schema: pkg
+      provider: hydra-colorlog
+    - path: ''
+      schema: structured
+      provider: schema
+    output_dir: /home/mathis/TEMOS/outputs/kit-mmm-xyz/baseline/3l49g7hv
+    choices:
+      callback: base
+      logger: none
+      sampler: variable_conseq
+      trainer: gpu
+      machine: server
+      model: temos
+      model/optim: adamw
+      model/losses: temos
+      model/motiondecoder: actor
+      model/motionencoder: actor
+      model/textencoder: distilbert_actor
+      data: kit-mmm-xyz
+      transforms/joints2jfeats@data.transforms.joints2jfeats: rifke
+      hydra/env: default
+      hydra/callbacks: null
+      hydra/job_logging: rich
+      hydra/hydra_logging: rich
+      hydra/hydra_help: default
+      hydra/help: default
+      hydra/sweeper: basic
+      hydra/launcher: basic
+      hydra/output: default
+  verbose: false

kit-mmm-xyz/3l49g7hv/.hydra/overrides.yaml

@@ -0,0 +1 @@
+[]

kit-mmm-xyz/3l49g7hv/checkpoints/last.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:99cba917da6068e0b59d23573d34c4a59fbbc493eb96573cb96db3c4e69331a5
+size 473558970