随笔分类 -  深度学习工具

摘要:问题: https://www.programcreek.com/python/example/90420/tensorflow.map_fn https://stackoverflow.com/questions/34987509/tensorflow-max-of-a-tensor-along- 阅读全文
posted @ 2019-01-14 21:36 乐乐章 阅读(614) 评论(0) 推荐(0)
摘要:TimeDistributed这个层还是比较难理解的。事实上通过这个层我们可以实现从二维像三维的过渡,甚至通过这个层的包装,我们可以实现图像分类视频分类的转化。 考虑一批32个样本,其中每个样本是一个由16个维度组成的10个向量的序列。该层的批输入形状然后(32, 10, 16)。 可以这么理解,输 阅读全文
posted @ 2018-05-24 16:36 乐乐章 阅读(3533) 评论(0) 推荐(0)
摘要:问题 在定义模型的时候,自定义了一个函数 模型保存之后,load 模型的时候报错: 解决 load 模型的时候需要指定custom object 参考: https://faroit.github.io/keras docs/2.1.5/getting started/faq/ handling c 阅读全文
posted @ 2018-05-24 16:17 乐乐章 阅读(1283) 评论(0) 推荐(0)
摘要:1. 如何进行迁移 对模型和相应的数据进行.cuda()处理。通过这种方式,我们就可以将内存中的数据复制到GPU的显存中去。从而可以通过GPU来进行运算了。 1.1 判定使用GPU 下载了对应的GPU版本的Pytorch之后,要确保GPU是可以进行使用的,通过torch.cuda.is_availa 阅读全文
posted @ 2018-05-16 20:12 乐乐章 阅读(17445) 评论(0) 推荐(0)
摘要:在运行上面的blog的Tensorflow小程序的时候程序我们会遇到一个问题,当然这个问题不影响我们实际的结果计算,但是会给同样使用这台计算机的人带来麻烦,程序会自动调用所有能调用到的资源,并且全占满,在自己的PC上没问题,但是在服务器上,问题就很大,因为一旦你运行程序,占满显存别人就不能再用了,解 阅读全文
posted @ 2018-05-16 20:08 乐乐章 阅读(1551) 评论(0) 推荐(1)