PyTorch-Reparam-Module:重新参数化你的 PyTorch 模块

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 }