2020年8月20日
使用pytorch的DistributedParallel进行单机多卡训练
摘要： 1. 导入库： import torch.distributed as dist from torch.utils.data.distributed import DistributedSampler 2. 进程初始化： parser = argparse.ArgumentParser() pars 阅读全文
posted @ 2020-08-20 16:44 Junzhao 阅读(5253) 评论(0) 推荐(0) 编辑