question about H36M model #25
Description
When I trained with mixed datasets, the best result was generated by 76th epoch. The best mPJPE is 56.21, and PAmPJPE is 35.62, which are different from (https://datarelease.blob.core.windows.net/metro/models/2021-02-25-graphormer_h36m_log.txt), MPJPE 51.2, PAmPJPE 34.5.
The command is the same as in EXP.md
python -m torch.distributed.launch --nproc_per_node=8
src/tools/run_gphmer_bodymesh.py
--train_yaml Tax-H36m-coco40k-Muco-UP-Mpii/train.yaml
--val_yaml human3.6m/valid.protocol2.yaml
--arch hrnet-w64
--num_workers 4
--per_gpu_train_batch_size 25
--per_gpu_eval_batch_size 25
--num_hidden_layers 4
--num_attention_heads 4
--lr 1e-4
--num_train_epochs 200
--input_feat_dim 2051,512,128
--hidden_feat_dim 1024,256,64
Looking forward to your reply @kevinlin311tw, thx