Quellcode durchsuchen

修改模型预测脚本

liyan vor 1 Jahr
Ursprung
Commit
f56afe400b
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      predict_pt.py

+ 1 - 1
predict_pt.py

@@ -15,7 +15,7 @@ parser.add_argument('--normalization', default='sigmoid', type=str, help='|选
 parser.add_argument('--batch', default=1, type=int, help='|输入图片批量|')
 parser.add_argument('--device', default='cuda', type=str, help='|推理设备|')
 parser.add_argument('--num_worker', default=0, type=int, help='|CPU处理数据的进程数,0只有一个主进程,一般为0、2、4、8|')
-parser.add_argument('--float16', default=True, type=bool, help='|推理数据类型,要支持float16的GPU,False时为float32|')
+parser.add_argument('--float16', default=False, type=bool, help='|推理数据类型,要支持float16的GPU,False时为float32|')
 args, _ = parser.parse_known_args()  # 防止传入参数冲突,替代args = parser.parse_args()
 # -------------------------------------------------------------------------------------------------------------------- #
 assert os.path.exists(args.model_path), f'! model_path不存在:{args.model_path} !'