![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
张先森啊
计算机专业
展开
-
将一个(768,1)的张量转换为(1024,)
我想要将一个(768,1)的张量转换为(1024,)试了下面几个方法都没有得到目标结果。最后通过零填充实现的。原创 2024-04-26 09:51:18 · 93 阅读 · 0 评论 -
RuntimeError: Expected to have finished reduction in the prior iteration before starting a new one.
在DistributedDataParallel中加入find_unused_parameters=True,如model = torch.nn.parallel.DistributedDataParallel(model, device_ids=[args.local_rank], output_device=args.local_rank, find_unused_parameters=True)这是因为部分参数没有参与计算loss。原创 2024-04-17 12:13:40 · 203 阅读 · 0 评论 -
什么是预训练
预训练是出于特殊数据集和低成本数据集数量差距悬殊而提出的一种模型训练方法,即先通过大量的数据集训练得到一组权重可以较好的识别这些数据的共性,从而进行分类、检测等操作,当我们需要处理自己特殊需求的任务时,我们就可以借助之前训练好的模型来针对我们的数据进行参数的调整,从而大大减少了数据集的需求和工作量。原创 2023-12-19 21:58:36 · 501 阅读 · 0 评论 -
船只图片资源以及标注
一共有**7000张船只图片**,标注的类别包括船只和海岸两类原创 2023-12-13 21:02:58 · 520 阅读 · 0 评论