- 博客(2)
- 收藏
- 关注
原创 pytorch 单gpu改为多gpu训练和测试
pytorch 单gpu改为多gpu训练和测试 训练中 在plfd_backbone = plfd_backbone.to(device)之前加 import torch.nn as nn plfd_backbone = nn.DataParallel(plfd_backbone, device_ids = [0, 1, 2, 3]) device_ids:设置gpu个数。 测试时,同样要修改 在to(device)后加入 pfld_backbone = nn.DataParallel(pfld_ba
2020-12-29 11:39:54 2357 1
原创 单向链表中输出倒数第k个节点
单向链表中输出倒数第k个节点 class ListNode: def __init__(self, val): self.val = val self.next = None def find_kth_to_tail(pHead, k): if not pHead: return None slow = pHead fast = pHead for i in range(1, k): if not fas
2020-12-02 13:49:33 130
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人