- 博客(2)
- 收藏
- 关注
RSS订阅原创 2021/9/17
import 日志 print（生活本就是一团糟，我们应当砥砺前行。 向往着光明，无所畏惧。生活总会好起来的。 孤独的日子总会让人进步，要牢记那时的奋斗。 现在想想，以前很多事情应该坚持，很多事情应该放弃，只是当时根本想不到那些事情会给现在的我带来什么。生活总是要充满遗憾的。 我觉得我自始至终都不是一个好人，也不算太坏。加油） ...
2021-09-17 19:22:52
60
原创 nn.DataParallel划分输入tensor的细节
写dataset时把多类型数据放到一起，但在模型中需要对不同类型数据有不同的支路。只用一块GPU没问题，直接按照数据组织方式从维度上把tensor切分就行了。当用两块GPU，不知道tensor的组织方式，无法按照维度索引进行切分，因此进行了详细的调试。 调试结果：nn.DataParallel将tensor在batch size维度上直接对半切分。直观看上去就是[1, 2, 3, 4]分成了[1, 2]和[3, 4]。 ...
2021-07-20 11:06:37
244
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人