Training poseclassification net with custom data

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Please provide the following information when requesting support.

• Hardware (T4/V100/Xavier/Nano/etc) RTX4080
• Network Type (Detectnet_v2/Faster_rcnn/Yolo_v4/LPRnet/Mask_rcnn/Classification/etc) poseclassificationnet
• TLT Version (Please run “tlt info --verbose” and share “docker_tag” here) nvcr.io/nvidia/tao/tao-toolkit:5.3.0-pyt

If I retrain with pretrained model “st-gcn_3dbp_nvidia.tlt”, I have size mismatch error.

size mismatch for fcn.weight: copying a param with shape torch.Size([6, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([4, 256, 1, 1]).
	size mismatch for fcn.bias: copying a param with shape torch.Size([6]) from checkpoint, the shape in current model is torch.Size([4]).

Customized data can’t use pretrained model?

3

If classes does not match, yes, will need to train from scratch.

4

There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks

5

Thank you. It is settled.

1 Like
6

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.