【文件属性】:
文件名称:PyTorch-Reparam-Module:重新参数化你的 PyTorch 模块
文件大小:8KB
文件格式:ZIP
更新时间:2021-05-31 05:31:42
Python
PyTorch-Reparam-Module
重新参数化你的 PyTorch 模块
要求
>= 1.2.0
Python 3
例子
import torch
import torch . nn . functional as F
import torchvision
from torchreparam import ReparamModule
device = torch . device ( 'cuda' )
# A regular network
net = torchvision . models . resnet18 (). to ( device )
# Reparametrize it!
reparam_net = ReparamModule ( net )
print ( f"reparam_net has { reparam_net . param_numel }
【文件预览】:
PyTorch-Reparam-Module-master
----.flake8(233B)
----torchreparam()
--------reparam_module.py(6KB)
--------__init__.py(42B)
----requirements.txt(15B)
----LICENSE(1KB)
----test()
--------test_reparam_module.py(4KB)
----setup.py(552B)
----README.md(4KB)
----.gitignore(1KB)