From ce42db0d7bbf344b34438d5bdd968e1a7753c758 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 17 Sep 2019 15:36:31 +0200 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 711510a6..054f2d09 100755 --- a/models.py +++ b/models.py @@ -88,7 +88,7 @@ def create_modules(module_defs, img_size, arc): b = [10, -0.1] elif arc == 'Fdefault': # Focal default no pw (28 cls, 21 obj, no pw) b = [-2.1, -1.8] - elif arc == 'uFBCE': # unified FocalBCE (5120 obj, 80 classes) + elif arc == 'uFBCE' or arc =='uFBCEpw': # unified FocalBCE (5120 obj, 80 classes) b = [0, -6.5] elif arc == 'uFCE': # unified FocalCE (64 cls, 1 background + 80 classes) b = [7.7, -1.1]