lrchang-优快云博客

原创 transformers库中的Trainer运行出现CUDA out of memory

假设你有 4 个 GPU，编号为 0, 1, 2, 3。如果你设置了 os.environ[“CUDA_VISIBLE_DEVICES”]=‘2’，那么程序会将编号为 2 的 GPU 映射为程序中的 GPU 0，从而只使用这一块 GPU。通过设置 os.environ[“CUDA_VISIBLE_DEVICES”]=‘2’，你的程序会认为只有编号为 2 的 GPU 存在，并且只会在这个 GPU 上运行计算任务。如果一个程序只需要一个 GPU，你可以指定一个特定的 GPU 而不是默认使用全部可用 GPU。

2024-08-17 10:07:43 678

原创【剑指offer】动态规划题目

【代码】【剑指offer】动态规划题目。

2022-08-24 17:59:47 304

原创【工具】python中字符串的知识整理ing

python字符串知识

2022-07-09 14:55:45 291

原创【机器学习系列】浙大机器学习课程-第二章支持向量机

支持向量机笔记

2022-07-02 13:08:55 570

原创【剑指offer】二叉树的题目

剑指offer二叉树题目解析

2022-07-01 22:39:36 544

原创【机器学习系列】浙大机器学习课程-第一章概述

机器学习入门概述

2022-07-01 20:38:56 627

原创【剑指offer】链表以及队列的题目

剑指offer链表及队列题目解析

2022-06-30 20:29:22 503

原创【深度学习系列】贝叶斯神经网络

如果一个算法对某些问题有效，那么它一定在另外一些问题比纯随机搜索方法更差。

2022-06-07 10:28:07 6149 1

原创【深度学习系列】反向传播推导+损失函数+激活函数原理

反向传播推导公式

2022-06-01 08:46:33 655

原创纵向联邦学习原理介绍——LR,XGBoost,SplitNN

作为一种保护隐私的分布式训练技术，联邦学习近几年成为学术界研究的热点话题，很多研究关于联邦学习的效率、安全性、隐私性展开讨论。最近在企业中联邦学习的应用也逐渐开展，常见的技术是基于FATE的联邦学习。

2022-05-28 20:21:46 5048 2

原创【联邦学习邂逅密码学系列】基于同态加密算法python代码实现

联邦学习是一种参与方之间联合隐私训练的新范式，受到学术界和工业界的关注。然而一些研究表明，联邦学习传输的中间信息，例如横向联邦学习中的梯度信息或者纵向联邦学习中的嵌入表示存在隐私泄露的风险。如何保密这些中间信息是一个重要研究问题。在这个系列中讨论如何将同态加密技术应用于联邦学习，保护参与方的数据隐私。

2022-05-21 09:53:45 11899 10

qq_42217215的博客