 05-23 14:26:26 | {'a_encoder': None,
 'a_fix_pre': False,
 'a_pre_encoder': None,
 'acc_weight': 0.0,
 'additional_data': False,
 'adv_weight': 20.0,
 'ali_weight': 0.0,
 'amsgrad': False,
 'apex': False,
 'asmr': 0.0,
 'atcont': 0.0,
 'atmr': 0.0,
 'aud_prob': 1.0,
 'audio_dims': 1,
 'audio_f': 256,
 'audio_fps': 16000,
 'audio_norm': False,
 'audio_rep': 'onset+amplitude',
 'audio_sr': 16000,
 'batch_size': 64,
 'beat_align': True,
 'benchmark': True,
 'cache_only': False,
 'cache_path': './datasets/beat_cache/id2/beat_smplx_en_emage/',
 'cap_rep': 'mxbai-embed-large',
 'caption': 'id2_caption',
 'cf': 0.0,
 'ch': 1.0,
 'cl': 1.0,
 'clean_final_seconds': 0,
 'clean_first_seconds': 0,
 'config': './configs/pyramotion.yaml',
 'csv_name': 'a2g_0',
 'cu': 1.0,
 'cudnn_enabled': True,
 'd_lr_weight': 0.2,
 'd_name': None,
 'data_path': './PyraMotion/BEAT2/beat_english_v2.0.0/',
 'data_path_1': './PyraMotion/pretrained/',
 'dataset': 'beat_sep_lower',
 'ddp': False,
 'debug': False,
 'decay_epochs': 9999,
 'decay_rate': 0.1,
 'decode_fusion': None,
 'deterministic': True,
 'device': 1,
 'disable_filtering': False,
 'div_reg_weight': 0.0,
 'dropout_prob': 0.3,
 'e_name': 'VAESKConv',
 'e_path': 'weights/AESKConv_240_100.bin',
 'emo_rep': None,
 'emotion_dims': 8,
 'emotion_f': 0,
 'epoch_stage': 0,
 'epochs': 500,
 'eval_model': 'motion_representation',
 'f_encoder': 'null',
 'f_fix_pre': False,
 'f_pre_encoder': 'null',
 'fac_prob': 1.0,
 'facial_dims': 100,
 'facial_f': 0,
 'facial_fps': 15,
 'facial_norm': False,
 'facial_rep': 'smplxflame_30',
 'fid_weight': 0.0,
 'finger_net': 'original',
 'freeze_wordembed': False,
 'fsmr': 0.0,
 'ftmr': 0.0,
 'fusion_mode': 'sum',
 'g_name': 'MAGE_Transformer',
 'gap_weight': 0.0,
 'gpus': [0],
 'grad_norm': 0.99,
 'hidden_size': 768,
 'id_rep': 'onehot',
 'input_context': 'both',
 'is_train': True,
 'ita_weight': 0.0,
 'iwa_weight': 0.0,
 'kld_aud_weight': 0.0,
 'kld_fac_weight': 0.0,
 'kld_weight': 0.0,
 'l': 4,
 'lf': 0.1,
 'lh': 3.0,
 'll': 3.0,
 'loader_workers': 0,
 'log_period': 10,
 'loss_contrastive_neg_weight': 0.005,
 'loss_contrastive_pos_weight': 0.2,
 'loss_gan_weight': 5.0,
 'loss_kld_weight': 0.1,
 'loss_physical_weight': 0.0,
 'loss_reg_weight': 0.05,
 'loss_regression_weight': 70.0,
 'lr_base': 0.0005,
 'lr_min': 1e-07,
 'lr_policy': 'step',
 'lu': 3.0,
 'm_decoder': None,
 'm_encoder': 'null',
 'm_fix_pre': False,
 'm_pre_encoder': 'null',
 'mean_pose_path': '/datasets/trinity/train/',
 'model': 'pyramotion',
 'momentum': 0.8,
 'motion_f': 256,
 'msmr': 0.0,
 'mtmr': 0.0,
 'multi_length_training': [1.0],
 'n_layer': 1,
 'n_poses': 34,
 'n_pre_poses': 4,
 'name': '0523_142626_pyramotion',
 'nesterov': True,
 'new_cache': False,
 'no_adv_epoch': 999,
 'notes': '',
 'opt': 'adam',
 'opt_betas': [0.5, 0.999],
 'ori_joints': 'beat_smplx_joints',
 'out_path': './PyraMotion/outputs/audio2pose/',
 'pos_encoding_type': 'sin',
 'pos_prob': 1.0,
 'pose_dims': 330,
 'pose_fps': 30,
 'pose_length': 64,
 'pose_norm': False,
 'pose_rep': 'smplxflame_30',
 'pre_frames': 8,
 'pre_type': 'zero',
 'pretrain': False,
 'project': 's2g',
 'queue_size': 1024,
 'random_seed': 2021,
 'rec_aud_weight': 0.0,
 'rec_fac_weight': 0.0,
 'rec_pos_weight': 0.0,
 'rec_txt_weight': 0.0,
 'rec_ver_weight': 0.0,
 'rec_weight': 1.0,
 'render_concurrent_num': 32,
 'render_tmp_img_filetype': 'bmp',
 'render_video_fps': 30,
 'render_video_height': 720,
 'render_video_width': 1920,
 'root_path': './PyraMotion/',
 'rot6d': True,
 'sem_rep': None,
 'sparse': 1,
 'speaker_dims': 30,
 'speaker_f': 0,
 'speaker_id': 'onehot',
 'split_rule': 'id2_train_test_split',
 'stat': 'ts',
 'std_pose_path': '/datasets/trinity/train/',
 'stride': 20,
 't_encoder': 'null',
 't_fix_pre': False,
 't_pre_encoder': 'fasttext',
 'tar_joints': 'beat_smplx_full',
 'test_ckpt': './PyraMotion/pretrained/last_420.bin',
 'test_data_path': '/datasets/trinity/test/',
 'test_length': 64,
 'test_period': 10,
 'train_data_path': '/datasets/trinity/train/',
 'train_trans': True,
 'trainer': 'pyramotion',
 'training_speakers': [2],
 'tsmr': 0.0,
 'ttmr': 0.0,
 'txt_prob': 1.0,
 'use_aug': False,
 'vae_codebook_size': 256,
 'vae_grow': [1, 1, 2, 1],
 'vae_layer': 4,
 'vae_length': 240,
 'vae_quantizer_lambda': 1.0,
 'vae_test_dim': 330,
 'vae_test_len': 32,
 'vae_test_stride': 20,
 'val_data_path': '/datasets/trinity/val/',
 'variational': False,
 'vel_weight': 0.0,
 'warmup_epochs': 0,
 'warmup_lr': 0.0005,
 'wei_weight': 0.0,
 'weight_decay': 0.0,
 'word_cache': False,
 'word_dims': 300,
 'word_f': 256,
 'word_index_num': 11195,
 'word_rep': 'textgrid',
 'z_type': 'speaker'}
 05-23 14:26:26 | # ------------ 0523_142626_pyramotion ----------- #
 05-23 14:26:26 | PyTorch version: 2.4.1+cu121
 05-23 14:26:26 | CUDA version: 12.1
 05-23 14:26:26 | 1 GPUs
 05-23 14:26:26 | Random Seed: 2021
 05-23 14:26:27 | An error has been caught in function '<module>', process 'MainProcess' (3166674), thread 'MainThread' (131719425808192):
Traceback (most recent call last):

> File "pyramotion_test.py", line 299, in <module>
    main_worker(0, 1, args)
    │                 └ Namespace(a_encoder=None, a_fix_pre=False, a_pre_encoder=None, acc_weight=0.0, additional_data=False, adv_weight=20.0, ali_we...
    └ <function main_worker at 0x77cb67397820>

  File "pyramotion_test.py", line 273, in main_worker
    trainer = __import__(f"{args.trainer}_trainer", fromlist=["something"]).CustomTrainer(

  File "/mnt/sda/william/AMG/PyraMotion/scripts/pyramotion_trainer.py", line 29, in __init__
    super().__init__(args)
                     └ Namespace(a_encoder=None, a_fix_pre=False, a_pre_encoder=None, acc_weight=0.0, additional_data=False, adv_weight=20.0, ali_we...

  File "/mnt/sda/william/AMG/PyraMotion/scripts/train.py", line 44, in __init__
    self.train_data = __import__(f"dataloaders.{args.dataset}", fromlist=["something"]).CustomDataset(args, "train")
    │                                                                                                 └ Namespace(a_encoder=None, a_fix_pre=False, a_pre_encoder=None, acc_weight=0.0, additional_data=False, adv_weight=20.0, ali_we...
    └ <pyramotion_trainer.CustomTrainer object at 0x77cb67392e50>

  File "/mnt/sda/william/AMG/PyraMotion/scripts/dataloaders/beat_sep_lower.py", line 57, in __init__
    self.smplx = smplx.create(
    │            │     └ <function create at 0x77cafd6ba430>
    │            └ <module 'smplx' from '/home/zhizhuoyin/anaconda3/envs/emagepy38/lib/python3.8/site-packages/smplx/__init__.py'>
    └ <dataloaders.beat_sep_lower.CustomDataset object at 0x77cb64af9be0>

  File "/home/zhizhuoyin/anaconda3/envs/emagepy38/lib/python3.8/site-packages/smplx/body_models.py", line 2410, in create
    raise ValueError(f'Unknown model type {model_type}, exiting!')

ValueError: Unknown model type , exiting!
