search

FabianFuchsML / se3-transformer-public

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

distributed training #15

Open yuanqidu opened 3 years ago

yuanqidu commented 3 years ago

Dear author,

Does SE(3)-Transformer support distributed training in Torch?

Thanks

FabianFuchsML commented 3 years ago

We never ran it on multiple GPUs, but I would hope that's not too much effort to implement :)