Skip to content

Training time and performance are different from the paper. #4

@dobaeky

Description

@dobaeky

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions