search

FabianFuchsML / se3-transformer-public

code for the SE3 Transformers paper: https://arxiv.org/abs/2006.10503
486 stars 67 forks source link

weights initialization #22

Closed pollytur closed 3 years ago

pollytur commented 3 years ago

Is their any way to specify the weights initialization strategy during the GSE3Res block creation?

FabianFuchsML commented 3 years ago

We don't have anything set up to control initialisation globally, but it's absolutely possible. You will probably need to go into the various modules that are used in the init of GSE3Res, such as G1x1SE3. There, you can set the initialisation strategy manually. Hope that helps!

pollytur commented 3 years ago

Thank you for a fast reply!)