3DPW-TEST:
  KEYPOINT_LIST:
  - 25
  - 26
  - 27
  - 28
  - 29
  - 30
  - 31
  - 32
  - 33
  - 34
  - 35
  - 36
  - 37
  - 43
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_new/3dpw-test/{000000..000012}.tar
  USE_HIPS: false
AIC-TRAIN:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled/aic-train/{000000..000104}.tar
  epoch_size: 200000
AIC-TRAIN-WMASK:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_wmasks/aic-train-vitpose/{000000..000104}.tar
  epoch_size: 200000
AVA-TRAIN-MIDFRAMES-1FPS:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled/ava-train-midframes-1fps/{000000..000092}.tar
  epoch_size: 200000
AVA-TRAIN-MIDFRAMES-1FPS-WMASK:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_wmasks/ava-train-midframes-1fps-vitpose/{000000..000092}.tar
  epoch_size: 200000
CMU-MOCAP:
  DATASET_FILE: /fsx/shubham/code/hmr2023/data/datasets/cmu_mocap.npz
COCO-TRAIN-2014:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled/coco-train-2014/{000000..000017}.tar
  epoch_size: 360000
COCO-TRAIN-2014-VITPOSE-REPLICATE-PRUNED12:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_wmasks/coco-train-2014-vitpose-replicate-pruned12/{000000..000044}.tar
  epoch_size: 45000
COCO-TRAIN-2014-WMASK:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_wmasks/coco-train-2014/{000000..000034}.tar
  epoch_size: 35000
COCO-TRAIN-2014-WMASK-PRUNED:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_wmasks/coco-train-2014-pruned/{000000..000017}.tar
  epoch_size: 18000
COCO-VAL:
  KEYPOINT_LIST:
  - 0
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled/coco-val/{000000..000000}.tar
H36M-TRAIN:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled/h36m-train/{000000..000156}.tar
  epoch_size: 314000
H36M-TRAIN-WMASK:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_wmasks/h36m-train/{000000..000312}.tar
  epoch_size: 314000
H36M-VAL-P2:
  KEYPOINT_LIST:
  - 25
  - 26
  - 27
  - 28
  - 29
  - 30
  - 31
  - 32
  - 33
  - 34
  - 35
  - 36
  - 37
  - 43
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_new/h36m-val-p2/{000000..000013}.tar
  USE_HIPS: true
H36M-VAL-P2-OPENPOSE:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_new/h36m-val-p2-openpose/{000000..000013}.tar
INSTA-TRAIN:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled/insta-train/{000000..001828}.tar
  epoch_size: 4000000
INSTA-TRAIN-WMASK:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_wmasks/insta-train-vitpose/{000000..001828}.tar
  epoch_size: 4000000
MPI-INF-TRAIN:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled/mpi-inf-train/{000000..000048}.tar
  epoch_size: 100000
MPI-INF-TRAIN-PRUNED:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_new/mpi-inf-train-pruned/{000000..00006}.tar
  epoch_size: 12000
MPII-TRAIN:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled/mpii-train/{000000..000004}.tar
  epoch_size: 100000
MPII-TRAIN-WMASK:
  TYPE: ImageDataset
  URLS: /fsx/shubham/data/hmr2023_data_shuffled_wmasks/mpii-train/{000000..000009}.tar
  epoch_size: 100000