PyTorch On Angel, arming PyTorch with a powerful Parameter Server, which enable PyTorch to train very big models.
164
stars
51
forks
source link
fix torch model size exception when parsing nodes without edges #22
Closed
yinhang-e5b0b9e888aa closed 3 years ago
在使用GraphSage做预测或者生成embedding时，会报The size of torch model exceeds the cores的异常。发现是在处理孤立节点时，没有放回TorchModel。经过多次（10次+）以上的测试验证，没有报错。