pytorch
friedrichor
研一学生
研究方向:Large Multimodal Models, Multimodal Dialogue.
Github: https://github.com/friedrichor
知乎: https://www.zhihu.com/people/friedrichor
知乎持续更新,欢迎关注
展开
-
如何冻结模型,避免 model.train() 改变模型部分模块
彻底冻结模型原创 2023-09-16 15:22:06 · 371 阅读 · 0 评论 -
使用 WeightedRandomSampler 解决数据样本不均衡的问题
WeightedRandomSampler 使用说明原创 2023-04-02 09:00:00 · 1088 阅读 · 0 评论 -
pytorch输出tensor张量时有省略号的解决方案(将tensor完整输出)
pytorch输出tensor张量时有省略号的解决方案(将tensor完整输出)原创 2022-07-02 11:52:55 · 5378 阅读 · 4 评论 -
使用 roboflow 将自己的数据集转化为 YOLOv5 所需的格式
roboflowYOLOv5原创 2022-05-04 17:52:28 · 5958 阅读 · 10 评论 -
如何查看自己电脑当前版本CUDA 可兼容的最高版本
Win键 + Q 打开搜索框搜索 NVIDIA控制面板(NVIDIA Control Panel)并打开点击 帮助 :系统信息点击 组件,在下图位置就能看到 CUDA的版本原创 2022-04-25 16:41:06 · 2270 阅读 · 0 评论 -
检验是否安装成功pytorch
能够检验是否成功安装pytorch, cuda, cudnn,并能输出GPU型号import torcha = torch.cuda.is_available()print(a)ngpu= 1device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")print(device)print(torch.cuda.get_device_name(0))print(torch.ra原创 2020-10-23 19:27:36 · 22474 阅读 · 0 评论