摘要: 前言 模型中常见的可训练层包括卷积层和线性层,这里将给出计算公式并在pytorch下进行验证。 计算模型的参数: import torch.nn as nn def cal_params(model: nn.Module): num_learnable_params = sum(p.numel() 阅读全文
posted @ 2024-12-18 17:10 片刻的自由 阅读(1908) 评论(0) 推荐(0)