config_pw3d_eft_3D.yaml

DEVICE : 'cuda'
SEED_VALUE : 4321              
LOGDIR : ''                            
EXP_NAME : 'pw3d_eft'   
DEBUG : True 
OUTPUT_DIR : 'results' 

DATASET_NAME : 'pw3d'  
ESTIMATOR : 'eft'      
BODY_REPRESENTATION : '3D'  
SAMPLE_INTERVAL : 10                  

SMPL_MODEL_DIR : "data/smpl/" 

# model config
MODEL :
  TYPE : 'network'
  NAME : '' 
# sampling setting
  SAMPLE_TYPE : 'uniform' 
  SLIDE_WINDOW_Q : 10
  DROPOUT : 0.1   

  ENCODER_RESIDUAL : True
  ENCODER_HEAD : 4
  ENCODER_TRANSFORMER_BLOCK : 5 
  ENCODER_EMBEDDING_DIMENSION : 12 

  DECODER : 'transformer'  
  DECODER_INTERP : 'linear' 
  DECODER_RESIDUAL : True
  DECODER_HEAD : 4
  DECODER_TRANSFORMER_BLOCK : 5 
  DECODER_EMBEDDING_DIMENSION : 12 
  DECODER_TOKEN_WINDOW : 5  


# training config
TRAIN :
  BATCH_SIZE : 256
  WORKERS_NUM : 0
  EPOCH : 70
  LR : 0.001
  LRDECAY : 0.95
  RESUME : None
  VALIDATE : True 
  USE_SMPL_LOSS : False
  USE_6D_SMPL : False
  PRE_NORM : True


# test config
EVALUATE:
  PRETRAINED : 'data/checkpoints/pw3d_eft_3d/checkpoint.pth.tar'
  ROOT_RELATIVE : True 
  SLIDE_WINDOW_STEP_Q : 1
  INTERP : 'linear'
  RELATIVE_IMPROVEMENT : False
  DENOISE : False

# loss config
LOSS :
  LAMADA : 2.5
  W_DENOISE : 1.0

VIS:
  INPUT_VIDEO_NUMBER : 8
  INPUT_VIDEO_PATH : 'data/videos/'
  OUTPUT_VIDEO_PATH : 'demo/'
  START : 0
  END : 1000