Closed zkaiWu closed 2 years ago
the current implementation only supports batch size 1. This is related to the GNN we used. If you want to train with multiple GPUs, you may need to change the data loader to Pytorch Geometric loader.
So if I what to train SGPN, how many GPU's memory I need. 12G is not enough, I what to ask if 32G is ok?
training with multiple GPU is not possible for the current code. If you want to train with SGPN you should set the maximum edge to 80. https://github.com/ShunChengWu/3DSSG/blob/a143511eb7024cbb347019de734e0e495659fdcd/config_example.json#L72
Sorry to bother you again There are som error when I used more than one gpu to train sgpn, but that is fine when I just use one gpu.