search

ORNL / HydraGNN

Distributed PyTorch implementation of multi-headed graph convolutional neural networks
BSD 3-Clause "New" or "Revised" License
68 stars 29 forks source link

Small fix PAINN Hanging Gradients #300

Closed RylieWeaver closed 4 weeks ago

RylieWeaver commented 4 weeks ago

Small fix here. Would cause an error with hanging gradients if only one convolutional layer.