hookandlook
hookandlook copied to clipboard
arsenyinfo
A library helping to gather stats and run checks during training deep learning models with Pytorch
Users may want to use custom metrics. At the same time, current metrics set may be excessive: e.g. L2 norm and mean value and correlated a lot.
``` In [1]: import torch In [2]: from hookandlook.watch import Wrapper In [3]: model = Wrapper.wrap_model(torch.nn.Linear(10, 10)) In [4]: torch.save(model, '/tmp/model.pth') --------------------------------------------------------------------------- AttributeError Traceback (most recent call last) in ---->...
Metadata
Owner
Metadata
A library helping to gather stats and run checks during training deep learning models with Pytorch