graph nn physics:图神经网络的可微物理模拟 源码

图神经网络的微分物理模拟 这是用PyTorch编写的“的实现。 目前,基本的2D模拟效果很好。 我没有时间训练模型超过40万个时间段,因此当前结果没有像将网络训练到20M个时间段时那样完美。 在给出初始位置和五个先前速度后,以下是网络预测的一些示例: 如您所见,到目前为止,网络还没有完全学会尊重边界,并且由于某些原因,远离其他粒子的粒子会以奇怪的方向飞走(这可能是由于网络对没有粒子的粒子缺乏经验)边缘连接到他们)。 但是,它具有正确的想法,并且大多数模拟看起来都是合理的。 我预计该网络可能无需修改就可以使用3D数据集,但是训练这样的网络需要比我可以使用的tesla v100或RTX 30