MgArcher/Text_select_captcha

你好,不知道能否提供训练代码呢

Closed this issue · 0 comments

你好,不知道能否提供给小白训练代码呢,用您提供的数据集使用torch yolov3官方代码训练得到pt模型,model.load_state_dict(
torch.load(opt.weights_path, map_location="cuda:0" if torch.cuda.is_available() and opt.GPU else "cpu")) 这个位置会出错,不知道是什么原因