centerpoint模型说明

本文主要是介绍centerpoint模型说明，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

1、调试时，执行到 for i, data_batch in enumerate(data_loader):时，第一次ok，第二次卡主了。网上给出的解决方法是将加载器DataLoader参数num_workers设为1

        batch_size = num_gpus * batch_size
        num_workers = num_gpus * workers_per_gpu

    # TODO change pin_memory
    data_loader = DataLoader(
        dataset,
        batch_size=batch_size,
        sampler=sampler,
        shuffle=(sampler is None),
        num_workers=num_workers,
        collate_fn=collate_kitti,
        # pin_memory=True,
        pin_memory=False,
    )

这篇关于centerpoint模型说明的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！