scglue.models.nn
Neural network modules, datasets & data loaders, and other utilities
Functions
Get torch computation device automatically based on GPU availability and memory usage |
|
Selectively stops normalization layers from updating running stats |
|
Get numpy dtype matching that of the pytorch default dtype |
Classes
Graph attention |
|
Graph convolution (propagation only) |