pytorch pin_memory

发表于 | 阅读次数:

pin_memory这样页面就不会被换出到硬盘。

1
2
3
4
5
6
test_loader = DataLoader(
    dataset=Human36M(actions=actions, data_path=opt.data_dir, use_hg=opt.use_hg, is_train=False),
    batch_size=opt.test_batch,
    shuffle=False,
    num_workers=opt.job,
    pin_memory=True)

请作者喝一杯咖啡☕️