pytorch
m0_51362950
这个作者很懒,什么都没留下…
展开
-
2021-09-10
torch.stack()input:list ,列表的元素是shape一样的tensor,加入是(x,y)dim: 叠加维度的方向,默认是0output:tensor ,其shape是(len(list),x,y)原创 2021-09-10 15:32:34 · 57 阅读 · 0 评论 -
2021-09-09
torchsummary.summary这是一个显示模型参数的方法,注意不是类,不需要创建对象,第一个参数是模型,即nn.module类,第二个是输入数据的shape,类型为元组,注意,不需要输入batch数,summary(model,input_data=(3,416,416))输入是三通道图像如果是输入下面的就是错误的形式summary(model,input_data=(batch_size,3,416,416))...原创 2021-09-09 21:44:20 · 72 阅读 · 0 评论 -
2021-09-09
argparse- actionparser.add_argument('-m','--tt',action='store_true')我的理解就是有一个名为‘tt’的参数,布尔类型,默认为false,当命令行参数出现-m的时候,其值改变为false。就是通过命令行改变其参数的布尔值。不知道还有其余深层次的理解不?...原创 2021-09-09 21:28:38 · 46 阅读 · 0 评论 -
2021-07-05
pytorch用于环境小知识点1、torch.cuda.is_available()判断cuda是否可用2、torch.device(“cuda:0”)返回一个device对象3、torch.cuda.DoubleTensor是加载在cuda的变量,torch.FloatTensor是加载cpu的变量4、model.cuda(device=device对象)代表model加载在device上5、tensor对象.to(device=device对象)代表tensor加载在device上6、mod原创 2021-07-05 21:44:54 · 63 阅读 · 0 评论