
Tensorflow/Pytorch
文章平均质量分 61
Hansry
我希望能有个明亮的落地窗,每天都晒一晒太阳。。。
展开
-
吴恩达Coursera深度学习课程 DeepLearning.ai 编程作业——Tensorflow+tutorial(2-3)
TensorFlow TutorialWelcome to this week’s programming assignment. Until now, you’ve always used numpy to build neural networks. Now we will step you through a deep learning framework that will allow yo原创 2017-11-21 22:37:45 · 7575 阅读 · 1 评论 -
Python 与 Tensorflow 简介
一. Python 的安装1. Window 平台安装 Python 打开 WEB 浏览器访问:https://www.python.org/downloads/windows/在下载列表中选择Window平台安装包,包格式为:python-XYZ.msi文件 , XYZ 为你要安装的版本号。下载后,双击下载包,进入Python安装向导,安装非常简单,你只需要使用默认的设置一直点击”...原创 2018-04-11 23:26:06 · 2026 阅读 · 0 评论 -
GPU的程序kill后未释放内存
使用PyTorch设置多线程(threads)进行数据读取(DataLoader),其实是假的多线程,他是开了N个子进程(PID都连着)进行模拟多线程工作,所以你的程序跑完或者中途kill掉主进程的话,子进程的GPU显存并不会被释放,需要手动一个一个kill才行,具体方法描述如下:1.先关闭ssh(或者shell)窗口,退出重新登录2.查看运行在gpu上的所有程序:fuser -v /dev...原创 2018-10-15 11:41:13 · 1249 阅读 · 0 评论 -
Pytorch 下使用 tensorboard 可视化
1. 安装tensorboardpip install tensorboardX2. 运行代码import torchimport torchvision.utils as vutilsimport numpy as npimport torchvision.models as modelsfrom torchvision import datasetsfrom tensorbo...原创 2018-10-15 14:58:43 · 2305 阅读 · 1 评论 -
远程访问服务器tensorboard
在登录远程服务器的时候使用命令:ssh -L 16006:127.0.0.1:6006 server_name@server.address其中server_name指工作站名字,server.address指工作站IP训练模型时使用如下命令:tensorboard --logdir="/path/to/log-directory"(其中,/path/to/log-direc...原创 2018-10-10 22:44:32 · 2361 阅读 · 0 评论 -
Torchvision transforms 总结
一.torchvision.transformsTransfoms 是很常用的图片变换方式,可以通过compose将各个变换串联起来**1. class torchvision.transforms.Compose (transforms) **这个类将多个变换方式结合在一起参数:各个变换的实例对象举例:transforms.Compose([ transforms.Center...原创 2018-11-16 17:00:04 · 26220 阅读 · 2 评论 -
关于pytorch grid_sample()
torch.nn.functional.grid_sample(input, grid, mode=‘bilinear’, padding_mode=‘zeros’)其中,grid的shape为(N, H, W, 2)对于output上的每一点,(x, y)三个通道的像素值,采集自input上某一点三个通道的像素值,采集的点存在于grid最低维,也就是(N, H, W, 2)中的2, [0]索...转载 2019-03-28 15:37:09 · 7190 阅读 · 0 评论