search

pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
20.54k stars 3.57k forks source link

Full repr string for `BatchNorm` #9390

Closed Kh4L closed 3 weeks ago

Kh4L commented 4 weeks ago

Rely on PyTorch's extra_repr to get more info in PyG's BatchNorm