Inconsistency between provided checkpoints and current model

dvlab-research / Stratified-Transformer

Stratified Transformer for 3D Point Cloud Segmentation (CVPR 2022)

MIT License

376 stars 40 forks source link

Inconsistency between provided checkpoints and current model #80

Closed databaaz closed 1 year ago

databaaz commented 1 year ago

The checkpoint file for Scannet (scannet_model_best.pth) has erroneous key names for the stem transformer layers. While the model is defined with the prefixes stem_layer...., the corresponding weights in the checkpoint are labelled with keys item_layer.... This affects 7 such weights in the checkpoint

X-Lai commented 1 year ago

Yes, you're correct. It is because of a typo when I run the experiment. You just need to replace all the item_layer with stem_layer as https://github.com/dvlab-research/Stratified-Transformer/blob/main/test.py#L104.