logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

UNet网络解读

UNet是一个对称的网络结构,左侧为下采样,右侧为上采样;下采样为encoder,上采样为decoder;四条灰色的平行线,就是在上采样的过程中,融合下采样过程的特征图的通道,Concat原理就是:一本大小为10cm10cm的书,厚度为3cm的书本(10103)的A书,和一本大小为10cm10cm,厚度为4cm的B书(10103)将A书和B书,边缘对齐的摞在一起,这样就可以得到一个大小10107的

#网络#深度学习
pytorch中.data()与.item(),.numpy(),.detach(),.cuda(),.cpu()的作用

我们在深度 pytorch 代码中经常看到以下表示,简单记录一下:.data:把变量设置成 tensor 类型,requires_grad 设置为 Flase。

#pytorch#numpy#python
PyTorch判断是否使用GPU以及哪些数据需要放到GPU上

输出若是 True,则 model 在 gpu 上;若是 False,则 model 在 cpu 上。输出 gpu 则在 gpu 上,输出 cpu 则在 cpu 上。

#pytorch#python#人工智能
    共 14 条
  • 1
  • 2
  • 请选择