Open cwj006 opened 1 year ago
Also, it would be better to install the latest version of pytorch and pyg, otherwise, it will take so so so long time for model.to(device).