深度学习中Python常用操作

cgnerds

已于 2022-03-26 20:44:10 修改

阅读量1k

点赞数

分类专栏：实用小程序文章标签： pytorch python

于 2022-03-25 09:21:22 首次发布

本文链接：https://blog.csdn.net/zhuxiaoyang2000/article/details/123726895

版权

实用小程序专栏收录该内容

8 篇文章 0 订阅

订阅专栏

文件夹下所有特定格式文件

查找 ‘./data/images’ 文件夹下所有 png格式图片。

import os

data_dir = './data/images'
filename = [name for name in os.listdir(data_dir) if os.path.splitext(name)[-1] == '.png']

模型参数大小

import torch
import torch.nn as nn

first_model = nn.Sequential(
                nn.Linear(3072, 512),
                nn.Tanh(),
                nn.Linear(512, 2),
                nn.LogSoftmax(dim=1))

sum([p.numel() for p in first_model.parameters()])