search

Nightmare-n / GD-MAE

GD-MAE: Generative Decoder for MAE Pre-training on LiDAR Point Clouds (CVPR 2023)
Apache License 2.0
114 stars 6 forks source link

pretrain weight for dla in Graph-VoI #10

Closed iMbaGong closed 1 year ago

iMbaGong commented 1 year ago

hi, thank for share your great work. i notice that in dla.py:

def dla34(pretrained=True, **kwargs):  # DLA-34
    model = DLA([1, 1, 1, 2, 2, 1],
                [16, 32, 64, 128, 256, 512],
                block=BasicBlock, **kwargs)
    return model

i can`t find any code to load the pretrain weight. how can i get and load the pretrain weight for dla?

Nightmare-n commented 1 year ago

The model would automatically load the pretrained weight of dla via this line according to the name of sub-modules.