Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
This is a minor change before first release targeted towards wider audience.
torchlayers.Infer
class is now torchlayers.infer
function and shouldn't be used as decorator as it destroys torch.build
due to pickle
error (see this issue for example).
Now, it should be used as a function as described in README.md
.Python 3.6
as torchlayers
relies on 3.7
module wide __getattr__
and __dir__
This is initial release of torchlayers
, project should be considered as being in alpha stage.
See README.md to get a grip on what this library does.
See documentation for in-depth description of provided layers & functionalities.