关于tensor的shape和dimension 刚把mnist又过了一遍,突然感觉对tensor的shape有了新的理解,虽然很基础,还是想分享一下 关于tensor的维度,也就是阶啦,从认知上理解...

Android 暗黑模式适配
Android 暗黑模式适配 不爱废话直接上菜。 第一种适配方式: 1.1 直接修改APP主题,values/styles.xml 如下: <style name="AppTheme" pare...
Tensorflow迁移学习加载预训练模型并修改变量shape和value的方法
迁移学习在深度学习中是经常被使用的方法,指的是在一个任务中预训练的模型被用于另一个任务的模型训练,以加快模型训练,减少资源消耗。 然而网络搜索相关的话题,基本上只涉及加载预训练模型的特定变量值的方法,...
Pytorch 类型错误:Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor.
Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor Pytorcht调试过程中,将数据传入模...
tensorflow根据输入更改tensor shape
涉及随机数以及类RNN的网络构建常常需要根据输入shape,决定中间变量的shape或步长。 tf.shape函数不同于tensor.shape.as_list()函数,后者返回的是常值list,而前...
Pytorch版本yolov3部分代码语句分析
这里只针对代码中出现的具体函数。使用pytorch版本的yolov3代码。 Pytorch版本的yolov3代码可见github连接:https://github.com/eriklindernore...

如何用pycharm来查看PyTorch,CUDA版本
如何用python来查看pycharm,CUDA版本 import torch print(torch.__version__) print(torch.version.cuda) 输出结果

安装Pytorch如何选择CUDA的版本
安装Pytorch时CUDA的选择 Nvidia CUDA 查看CUDA版本方法 查看CUDA的驱动API版本 查看CUDA的运行API版本 查看官方CUDA的运行API版本 方法1 方法2 查看An...
keras实现多gpu训练,单gpu保存模型,单gpu预测
问题:我在训练模型的时候用的是4个gpu,我需要将训练好的模型加载到树莓派中,但是树莓派只有一个cpu or gpu 解决方法:多gpu进行训练,单gpu保存训练的模型,用单gpu进行预测 代码呈现,...

【PyTorch】Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
和网上找到的答案都不一样。 问题是因为我在__init__外面的函数中定义了一个卷积操作 然后就一直报错。 具体原因之后再查吧。