Model Loader for PyTorch, Input Name & Output Model
pip install torch-model-loader==0.1.0
Model Loader for PyTorch, Name
Need Python 3.6+.
pip install torch-model-loaderimport torch.nn as nn
import torch_model_loader
@torch_model_loader.register
class CustomModule(nn.Module):
pass
# ...
model_class = torch_model_loader.load('CustomModule')
model = model_class()
The Tidelift Subscription provides access to a continuously curated stream of human-researched and maintainer-verified data on open source packages and their licenses, releases, vulnerabilities, and development practices.
Learn more →
Something wrong with this page? Make a suggestion
Export .ABOUT file for this package
Last synced: 2019-04-28 08:13:52 UTC
Login to resync this project