cwj006

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).

thanks