随笔分类 -  小土堆Pytorch

摘要:Pytorch学习资料:http://www.feiguyunai.com/index.php/2019/09/11/pytorch-char03/ ———————————————————————————————————— torch.nn Containers是神经网络骨架,含6个类,最常用的是M 阅读全文
posted @ 2024-08-04 16:37 xjl_ultrasound 阅读(157) 评论(0) 推荐(0)
摘要:学会看内置数据集的官方文档:https://pytorch.org/vision/stable/generated/torchvision.datasets.CIFAR10.html#torchvision.datasets.CIFAR10 示例代码: import torchvision from 阅读全文
posted @ 2024-08-03 11:33 xjl_ultrasound 阅读(55) 评论(0) 推荐(0)
摘要:transforms对图像进行改造 from torchvision import transforms 最靠谱的办法:根据help文件自行学习transforms包含哪些工具(类)以及如何使用 ———————————————————————————————————— 自学一个类时,应关注: 1、如 阅读全文
posted @ 2024-08-02 15:53 xjl_ultrasound 阅读(65) 评论(0) 推荐(0)
摘要:Tensorboard数据可视化 TensorBoard是一个可视化工具,它可以用来展示网络图、张量的指标变化、张量的分布情况等。它通过运行一个本地服务器,来监听6006端口(可更改)。在浏览器发出请求时,分析训练时记录的数据,绘制训练过程中的图像 当前环境下安装:pip install tenso 阅读全文
posted @ 2024-08-02 14:04 xjl_ultrasound 阅读(71) 评论(0) 推荐(0)
摘要:Dataset类 作用:模型的数据集接口 __init__将对象实例化,创建对象时obj = class(..., ...)会立即被调用,需要提供(输入)类中使用到的变量。 __getitem__通过img, label = obj[idx]获取(返回)每一个数据和label __len__通过le 阅读全文
posted @ 2024-08-01 18:25 xjl_ultrasound 阅读(78) 评论(0) 推荐(0)
摘要:Pytorch环境安装及配置 1、创建conda环境,名为pytorch conda create -n pytorch python=3.10 2、在任务管理器的性能中确认显卡,是否支持CUDA。其次,确认显卡驱动,cuda9.2支持396.26以上的驱动,可以在命令行输入nvidia-smi来看 阅读全文
posted @ 2024-08-01 17:14 xjl_ultrasound 阅读(150) 评论(0) 推荐(0)