search

IIT-PAVIS / DiffAssemble

Official repository for "DiffAssemble: A Unified Graph-Diffusion Model for 2D and 3D Reassembly" accepted at CVPR2024
66 stars 5 forks source link

how to train equivariant encoder #2

Closed zc1213856 closed 5 months ago

zc1213856 commented 8 months ago

Thanks for your great work! But I can't find a pre-trained equivariant encoder mentioned in the paper. And it seems that its weights are not frozen during the training phase (i run puzzle_diff\train_3d.py). Can you provide more detail about how to train this encoder?

gianscarpe commented 5 months ago

Hi and thanks for your interest in our work! That's expected, we don't pre-train the equivariant encoder but rather train it from scratch along with the despiser