深度学习中Python常用操作_深度学习训练python操作

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

查找 ‘./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']
1
2
3
4

模型参数大小

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()])
1
2
3
4
5
6
7
8
9
10