diff --git a/utils/run_task2.py b/utils/run_task2.py index a1dc6abb045938cf18cc0df63f45c4aae11839a0..3956f073a33caa4d2bf673e7fdb0826bc800184d 100755 --- a/utils/run_task2.py +++ b/utils/run_task2.py @@ -40,12 +40,12 @@ CONFIG = { "product_brand": 10643, "product_color_name": 11890, }, - "model": ["/models/us-cocolm-des-0-fp16.onnx", "/models/us-cocolm-des-0-fp16.onnx"], + "model": ["/models/us-cocolm-new-seed-0-fp16.onnx", "/models/us-cocolm-new-seed-1-fp16.onnx"], "product": "/models/cocolm.h5", "key": "us", "type": "cocolm", "fold_offset": 2, - "description":True, + "description":False, }, { "clean": DeBertaCleanV2, diff --git a/utils/run_task3.py b/utils/run_task3.py index 52ac662ac32c99739c5429feee054bb5fe04c100..f5bf0b4fd0ae484569ecc26a18499a72f87d6b54 100755 --- a/utils/run_task3.py +++ b/utils/run_task3.py @@ -39,12 +39,12 @@ CONFIG = { "product_brand": 10643, "product_color_name": 11890, }, - "model": ["/models/us-cocolm-des-0-fp16.onnx", "/models/us-cocolm-des-0-fp16.onnx"], + "model": ["/models/us-cocolm-new-seed-0-fp16.onnx", "/models/us-cocolm-new-seed-1-fp16.onnx"], "product": "/models/cocolm.h5", "key": "us", "type": "cocolm", "fold_offset": 2, - "description":True, + "description":False, }, { "clean": DeBertaCleanV2,