-
Notifications
You must be signed in to change notification settings - Fork 4
Description
I trained your model, 2G-GCN.
The model code is the same and the configure file is almost identical.
I trained this following process.
First, I trained 2G-GCN_stage1 on mphoi.
Second, I trained 2G-GCN_stage2 on mphoi.
The only difference is that when training stage1, I replaced "2G-GCN" with "2G-GCN_stage1" in conf/config.yaml and when training stage2, I replaced "2G-GCN_stage1" with "2G-GCN_stage2" in conf/config.yaml again.
Also, when training stage2, I substituted "pretrained_path: ${env:PWD}/outputs/cad120/2G-GCN/hs512_e50_bs16_lr0.0001_0.5_Subject1" with the checkpoint name of the output of stage1 in conf/models/2G-GCN_stage2.yaml.
And the rest of the settings are the same.
As a result, it trained well but the training time took only 1 hour on one RTX3090 GPU and I tested the output checkpoint by command " python -W ignore predict.py --pretrained_model_dir ./outputs/mphoi/2G-GCN/hs512_e40_bs8_lr0.0001_0.5_Subject45 --cross_validate".
The performance was as follows.
Summary F1@k results.
sub-activity_prediction
Overlap: 0.1
Values: [0.8852]
Mean: 0.8852 Std: 0.0000
Overlap: 0.25
Values: [0.863]
Mean: 0.8630 Std: 0.0000
Overlap: 0.5
Values: [0.6912]
Mean: 0.6912 Std: 0.0000
sub-activity_recognition
Overlap: 0.1
Values: [0.7372]
Mean: 0.7372 Std: 0.0000
Overlap: 0.25
Values: [0.6793]
Mean: 0.6793 Std: 0.0000
Overlap: 0.5
Values: [0.5447]
Mean: 0.5447 Std: 0.0000
I think that these results are too different from those reported in your paper.
I wonder if I'm doing it right.