ORNL / HydraGNN

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

added sigmoid activation function #254

Closed erdemcaliskan closed 3 months ago

erdemcaliskan commented 3 months ago

Sigmoid activation function option is added to model.py

allaffa commented 3 months ago

@erdemcaliskan Please use black to fit the formatting of your changes.

erdemcaliskan commented 3 months ago

Will do, thank you!