From 7a1d066fdc5160672f200c19d731d85e520ff0c5 Mon Sep 17 00:00:00 2001 From: Brainway Date: Wed, 28 Sep 2022 04:04:29 +0000 Subject: [PATCH] update predict.py. --- predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predict.py b/predict.py index a9612f2..b9b6910 100755 --- a/predict.py +++ b/predict.py @@ -17,7 +17,7 @@ def parse_args(): parser.add_argument('--split', type=str, default='overlap', help="Split method") # non-overlap parser.add_argument('--slide_step', type=int, default=2, help="Slide step for overlap split") parser.add_argument('--smoothing_value', type=float, default=0.0, help="Label smoothing value\n") - parser.add_argument("--pretrained_model", type=str, default="output/ieemooempty_checkpoint_good.pth", help="load pretrained model") + parser.add_argument("--pretrained_model", type=str, default="../module/ieemoo-ai-isempty/model/now/emptyjudge5_checkpoint.bin", help="load pretrained model") return parser.parse_args()