pytorch
Du_JuneLi
从事cv领域,无人驾驶感知、OCR等,包括检测、识别、分割、跟踪。
展开
-
表格识别-pytorch
任务介绍:img2html;基础模型:RARE;输入:表格图片(检测到的表格);输出:html;需要的朋友可以去我的GitHub,觉得不错的话点个star,ღ( ´・ᴗ・` );...原创 2022-01-31 17:31:30 · 2109 阅读 · 1 评论 -
pytorch并行相关
多GPU使用:CUDA_VISIBLE_DEVICES=0,1,2,3---推荐os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3"---推荐DataParallel(model, device_ids=[0, 1, 2,3])torch.cuda.set_device(id)---不推荐cuda:0表示将id为0的GPU设置为主GPU;相关链接:RuntimeError: module must have its parameters a原创 2021-11-16 18:47:06 · 2138 阅读 · 0 评论 -
pytorch常见问题
1. pytorch 的 dataloader 在读取数据时,设置了较大的 batchsize 和 num_workers. 然后训练一段时间报错:RuntimeError: Too many open files. Communication with the workers is no longer possible. Please increase the limit using ulimit -n in the shell or change the sharing strategy by c原创 2021-11-16 14:43:06 · 2869 阅读 · 0 评论 -
torch.optim相关
optim:torch.optim.SGD([ {'params': model.base.parameters()}, {'params': model.classifier.parameters(), 'lr': 1e-3}], lr=1e-2, momentum=0.9)Lambda LR:torch.optim.lr_scheduler.LambdaLR(optimizer,lr_lambda,last_epoch=-1)原创 2021-11-03 18:08:02 · 74 阅读 · 0 评论