
Torch
ZQ_ZHU
这个作者很懒,什么都没留下…
展开
-
PackagesNotFoundError: The following packages are not available from current channels:
错误信息如下:Collecting package metadata (current_repodata.json): doneSolving environment: failed with initial frozen solve. Retrying with flexible solve.Collecting package metadata (repodata.json): doneSolving environment: failed with initial frozen solve.原创 2020-10-14 14:44:13 · 1339 阅读 · 0 评论 -
pytorch快速安装【清华源】
step1:安装anacoda 下载地址bash Anaconda3-2018.12-Linux-x86_64.shstep2:按照官网的方法安装pytorchconda install pytorch torchvision cudatoolkit=10.0 -c pytorch在此期间你会发现特别的慢,慢也就算了主要它还老安装不成功,出现的错误见我的另一篇博客补充说明:...原创 2019-02-28 19:14:14 · 192335 阅读 · 94 评论 -
PyTorch创建自己的数据集
图片文件在同一的文件夹下思路是继承 torch.utils.data.Dataset,并重点重写其 __getitem__方法,示例代码如下:class ImageFolder720p(Dataset): """ Image shape is (720, 1280, 3) --> (768, 1280, 3) --> 6x10 128x128 patc原创 2019-03-10 20:00:11 · 1977 阅读 · 0 评论 -
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace
Traceback (most recent call last): File "train.py", line 182, in <module> train(config) File "train.py", line 120, in train loss.backward() File "/home/zqzhu/anaconda3/lib/python3....原创 2019-03-30 16:06:59 · 8972 阅读 · 0 评论 -
tf.einsum—爱因斯坦求和约定
1. einsum记法如果你像我一样,发现记住PyTorch/TensorFlow中那些计算点积、外积、转置、矩阵-向量乘法、矩阵-矩阵乘法的函数名字和签名很费劲,那么einsum记法就是我们的救星。einsum记法是一个表达以上这些运算,包括复杂张量运算在内的优雅方式,基本上,可以把einsum看成一种领域特定语言。一旦你理解并能利用einsum,除了不用记忆和频繁查找特定库函数这个好处...转载 2019-04-08 20:53:01 · 8031 阅读 · 2 评论 -
PyTorch基本数据类型及其占用空间大小
PyTorch上的常用数据类型如下Data typedtypeCPU tensorGPU tensorSize/bytes32-bit floatingtorch.float32 or torch.floattorch.FloatTensortorch.cuda.FloatTensor464-bit floatingtorch.float64 or t...原创 2019-04-23 15:38:58 · 11278 阅读 · 1 评论 -
保存tensor至本地文件
由于实验过程需要,将中间结果tensor保存至本地文件,这里利用numpy库中的savez函数来实现,具体的代码如下# 首先将要保存的tensor从计算图中分离出来,这里用到detach函数enc = by_latent.detach().cpu().numpy()因为用的是savez函数来实现保存,所以直接拿来用就可以,保存的默认文件后缀是*.npznp.savez("/test/en...原创 2019-05-21 20:29:02 · 13506 阅读 · 2 评论