Macro


  • 首页

  • 分类

  • 标签

  • 归档

  • 关于

  • 搜索

利用tensorwatch对pytorch网络可视化及统计参数

发表于 2019-06-16 | 分类于 ML/DL

代码

1
2
3
4
5
6
7
8
import sys
import torch
import tensorwatch as tw
import torchvision.models

alexnet_model = torchvision.models.alexnet()
tw.draw_model(alexnet_model, [1, 3, 224, 224])
tw.model_stats(alexnet_model, [1, 3, 224, 224])

结果

# pytorch # tensorwatch
pytorch实现迁移学习
pytorch中的一些技巧-统计参数数量,划分数据集,初始化,正则化
  • 文章目录
  • 站点概览
Macro

Macro

271 日志
7 分类
221 标签
  1. 1. 代码
  2. 2. 结果
© 2020 Macro