- 博客(6)
- 资源 (1)
- 收藏
- 关注
原创 Windows上使用PyTorch训练模型GPU利用率低的原因
在Windows上使用PyTorch训练模型时,任务管理器显示的GPU占用率可能较低,这并不准确。建议使用nvidia-smi dmon命令查看真实的GPU占用率,重点关注流处理器利用率(sm)和显存带宽利用率(mem),确保GPU满载。若GPU占用率确实低,可以通过调整PyTorch DataLoader的num_workers和pin_memory参数来优化。
2025-05-08 21:23:47
1115
原创 PyTorch使用生成对抗网络(GAN)生成手写数字源码(DCGAN、cGAN、WGAN-GP)
本文一共使用了四种方式实现了使用生成对抗网络(GAN)生成手写数字。首先我们使用了一个最简单的线性网络,遇到了模式崩溃问题。然后我们用卷积对抗网络(DCGAN),解决了生成手写数字问题。然后我们使用条件对抗网络(cGAN)实现了对生成数字的控制。最后我们使用了WGAN-GP网络,实现了比较稳定的训练。
2025-05-07 19:59:19
987
1
原创 使用PyTorch实现猫狗分类Python源码及准确度对比(CNN、VGG16迁移学习两张方式)
【代码】使用PyTorch实现猫狗分类Python源码及准确度对比(CNN、VGG16迁移学习两张方式)
2025-05-06 20:02:02
746
原创 XMind思维导图最后一个不限制画布数量的版本
XMind是一个全功能的思维导图和头脑风暴软件。最新版本只能创建3个画布,最后一个提供无限画布的版本是2024年6月发布的24.04.10311版本。
2025-05-06 19:37:00
521
原创 使用Ollama实现工具调用的原理及Python代码实现
这里以支持工具调用的qwen3为例,介绍实现原理。body模型返回tool_calls消息,要求调用get_current_weather工具。body模型返回结果。
2025-05-05 11:29:09
470
原创 使用PyTorch实现手写数字识别(Linear、CNN、Visual Transformer三种方式)
【代码】使用PyTorch实现手写数字识别(Linear、CNN、Visual Transformer三种方式)
2025-05-05 08:16:53
423
1
ShadowEditor-v0.5.2源码
2020-05-31
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人