android 黑夜模式适配记录 编程语言

android 黑夜模式适配记录

文章目录 前言 一、不开启深色模式 1.关闭自动深色模式 二、自己适配深色模式 1.关闭自动深色模式 2.创建res深色适配目录 (1) 适配colors,strings (2) 适配图片或者shap...
阅读全文
pytorch数据批次的大小对模型拟合效果的影响 编程语言

pytorch数据批次的大小对模型拟合效果的影响

本次使用大小为 的数据集进行回归预测,研究设置不同的数据批次对模型拟合效果的影响。 过程: 1、导入数据 2、建立 XGBRegressor() 模型,计算均方误差。 3、建立4个结构相同的深度学习模...
阅读全文

PyTorch查看CUDA支持情况

import torch import torchvision print(torch.cuda.is_available()) 上面的命令只是检测CUDA是否安装正确并能被Pytorch检测到,并没...
阅读全文
检查pytorch是否安装成功、查看torch和cuda的版本 编程语言

检查pytorch是否安装成功、查看torch和cuda的版本

方式 1 在安装完成pytorch和cudatoolkit之后,需要检查一下是否安装成功,可以直接查看当前环境下安装的所有包来验证: conda list 可以看到pytorch版本和cuda的版本分...
阅读全文
Pytorch单GPU、多GPU训练的几个细节对比 编程语言

Pytorch单GPU、多GPU训练的几个细节对比

Pytorch单GPU、多GPU训练的区别主要在三个地方:训练前指定GPU、训练过程中保存模型和加载刚刚保存的模型。 训练前指定GPU 单GPU: os.environ = '0' 多GPU: os....
阅读全文

pytorch多GPU训练保存的模型,在单GPU环境下加载出错

背景 在公司用多卡训练模型,得到权值文件后保存,然后回到实验室,没有多卡的环境,用单卡训练,加载模型时出错,因为单卡机器上,没有使用DataParallel来加载模型,所以会出现加载错误。原因 Dat...
阅读全文

TensorFlow之张量的形状相关操作

  TensorFlow之张量的形状相关的操作函数主要讲解一下四种。 1、tf.shape 查看张量的形状,即返回一个一阶张量表示输入张量的形状。 tf.shape(input, name=...
阅读全文
安卓夜间模式开发【深色主题适配】 编程语言

安卓夜间模式开发【深色主题适配】

1.创建文件夹 就在res 下新建 values-night目录,创建对应的colors.xml文件。如果适配图片的话,就创建对应的 drawable-night目录。  正常情况下,系统会...
阅读全文
pytorch的多GPU训练方法 编程语言

pytorch的多GPU训练方法

pytorch中共有两种多GPU的训练方法,一种是利用nn.DataParallel实现,实现简单,另一种是用采用分布式并行训练DistributedDataParallel和DistributedS...
阅读全文