相关文章
Pruning 系列 (十)Pruning VGG 实战
环境 python 3.9numpy 1.24.1pytorch 2.0.0+cu117流程 一、 VGG 模型代码
# -*- coding: utf-8 -*-
import torch
import torch.nn as nndefault_cfg = {11: [64, M, 128, M, 256, 256, M, 512, 512, M, 512, 512],13: [64, 64, M, 128, 128, M, 256, 256, M, 512, 512, M, …
建站知识
2024/12/24 2:37:37