摘要: Dataset类 作用:模型的数据集接口 __init__将对象实例化,创建对象时obj = class(..., ...)会立即被调用,需要提供(输入)类中使用到的变量。 __getitem__通过img, label = obj[idx]获取(返回)每一个数据和label __len__通过le 阅读全文
posted @ 2024-08-01 18:25 xjl_ultrasound 阅读(96) 评论(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 阅读(164) 评论(0) 推荐(0)