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 #298

Closed RylieWeaver closed 3 weeks ago

RylieWeaver commented 3 weeks ago

Small fix here, it would have caused errors of hanging gradients when there's only one convolutional layer.