文件名称：PyTorch-Reparam-Module:重新参数化你的 PyTorch 模块
文件大小：8KB
文件格式：ZIP
更新时间：2024-06-19 23:18:22
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)