- 博客(10)
- 收藏
- 关注
原创 分类网络中的top1和top5
import torchvisionimport torchimport torch.utils.data.dataloader as dataloaderimport torchvision.datasets as datasetsimport torchvision.transforms as transformsimport torch.nn as nnimport torch.optim as optim# 加载数据和处理数据transform = transforms.Comp.
2020-05-21 07:42:26 7735 4
原创 深度学习模型的保存和加载
保存和加载模型关于保存和加载模型,有三个核心功能需要熟悉:torch.save:将序列化的对象保存到磁盘。此函数使用Python的pickle实用程序进行序列化。使用此功能可以保存各种对象的模型,张量和字典。 torch.load:使用pickle的解码功能将序列化的目标文件反序列化到内存中。 torch.nn.Module.load_state_dict:使用反序列化的state_dict加载模型的参数字典 。什么是state_dict?在PyTorch中,模型的可学习参数(即权重和偏.
2020-05-29 17:29:26 5330
原创 深度学习常见代码汇总
1.选择要用的GPU: CUDA_VISIBLE_DEVICES服务器中有多个GPU,选择特定的GPU运行程序可在程序运行命令前使用:CUDA_VISIBLE_DEVICES=0命令。0为服务器中的GPU编号,可以为0, 1, 2, 3等,表明对程序可见的GPU编号。2.os.environ设置系统环境变量 :os.environ[‘环境变量名称’]=‘环境变量值’获取系统环境变量:os.environ[‘环境变量名称’]其中 key 和 value 均为 string 类型查看..
2020-05-29 09:23:13 645
原创 Python中的 isinstance() 函数简介和使用效果
Python中的 isinstance() 函数,是Python中的一个内置函数,用来判断一个函数是否是一个已知的类型。isinstance(object,classinfo)参数:object : 实例对象。classinfo : 可以是直接或者间接类名、基本类型或者由它们组成的元组。返回值:如果对象的类型与参数二的类型(classinfo)相同则返回 True,否则返回 False。3. 实例以下是使用isinstance()函数的实例:a = 15isinstance(a,i
2020-05-28 16:39:18 3268
原创 pycharm执行有输入参数的脚本
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter) # ************************************************************ # Datasets (general) # ************************************************************ pa.
2020-05-28 11:33:10 801
原创 解读resnet50的预训练权重.pth文件
参考了:https://blog.csdn.net/qq_27825451/article/details/100773473?ops_request_misc=&request_id=&biz_id=102&utm_term=%E5%A6%82%E4%BD%95%E6%9F%A5%E7%9C%8B.pth%E6%96%87%E4%BB%B6&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaidu
2020-05-24 09:02:37 10651
原创 使用Anaconda时能够新手简单操作命令
1. #列出已安装的包conda list2.查看CUDA版本nvcc --version3.使用conda卸载Pytorchconda uninstall pytorch4.使用pip卸载Pytorchpip uninstall pytorch5.pytorch安装可到官网https://pytorch.org/自己选择。
2020-05-23 13:04:02 186
原创 改变argparse linux形式手动输入参数
#!/usr/bin/env python3# -*- coding: utf-8 -*-""" @Author hekun @Date 2020/5/16 17:20"""#from __future__ import print_function # 这个是python当中让print都以python3的形式进行print,即把print视为函数import argparse # 使得我们能够手动输入命令行参数,就是让风格变得和Linux命令行差不多import tor.
2020-05-22 08:41:32 332
原创 运行.sh文件时遇到wget:command notf ound问题解决办法
可以通过以下两种方法来安装:1、rpm 安装rpm 下载源地址:http://vault.centos.org/centos/6.4/os/x86_64/Packages/下载wget的RPM包:http://vault.centos.org//centos/6.4/os/x86_64/Packages/wget-1.12-1.4.el6.x86_64.rpmrpm ivh wget-1.12-1.4.el6.x86_64.rpm 安装即可。2、yum安装yum -y insta..
2020-05-21 20:11:11 362
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人