4 projects
wattaged
wattaged is a utility that logs the consummed electricity and allows you to quickly compute the consummed kWh
torch-pconv
Faster and more memory efficient implementation of the Partial Convolution 2D layer in PyTorch equivalent to the standard NVidia implem.
pshape
Prints NumPy-like arrays' shapes, mins, means, and maxes, as well as the names of the input variable outside the functions' scope
tensor-type
Annotates shapes of PyTorch Tensors using type annotation in Python3, and provides optional runtime shape validation.