Description:
pytorchnntools 0.3.7
Tools for NN creation with Pytorch. The project is greatly inspired by fast.ai library.
The project brings the convenience functions from fast.ai to a usual pytorch user.
Usage examples
CheckpointSaver
from pytorch_nn_tools.devices import to_device
from pytorch_nn_tools.train.checkpoint import CheckpointSaver
checkpoint_saver = CheckpointSaver(path_checkpoints, logger=DummyLogger())
# create your model, optimizer, scheduler
# train for several epochs
for epoch in range(100):
.....
# at the end of each epoch we save the results
checkpoint_saver.save(model, optimizer, scheduler, epoch)
# later you can load the latest checkpoint and continue training
last = checkpoint_saver.find_last(start_epoch, end_epoch)
if last is not None:
print(f"found pretrained results for epoch {last}. Loading...")
self.checkpoint_saver.load(model, optimizer, scheduler, last)
else:
print(f"pretrained weights are not found")
Misc
Free software: MIT license
Documentation: https://pytorch-nn-tools.readthedocs.io.
License
For personal and professional use. You cannot resell or redistribute these repositories in their original state.
Customer Reviews
There are no reviews.
