After training for several epochs, add refine module?

[haofeng0705]

is it refer to Exploiting Spatial-temporal Relationships for 3D Pose Estimation via Graph Convolutional Networks, the refinement module? how many epochs do I need to train to add this module？

[Vegetebird]

Our repo is built on top of ST-GCN, we train the first stage for 20 epochs.

[haofeng0705]

thank you! btw, how can I save the predicted 3d poses and corresponding labels output from the model?

[Vegetebird]

You can save the outputs from the model (post_out) in https://github.com/Vegetebird/StridedTransformer-Pose3D/blob/6773a98f873f0d7bc6d8d0f6c0b1a478409e535e/demo/vis.py#L222