Update: model path

This commit is contained in:
dx-tan 2024-05-02 16:26:58 +07:00
parent a018f2bd0c
commit a76841cf4e

View File

@ -29,7 +29,7 @@ kvu_model = {
"option": "sbt_v2", "option": "sbt_v2",
"model": { "model": {
"pretrained_model_path": "/workspace/cope2n-ai-fi/weights/layoutxlm-base", "pretrained_model_path": "/workspace/cope2n-ai-fi/weights/layoutxlm-base",
"config": "/workspace/cope2n-ai-fi/weights/models/sdsvkvu/key_value_understanding_for_sbt-20240416-153954/base.yaml", "config": "/workspace/cope2n-ai-fi/weights/models/sdsvkvu/key_value_understanding_for_sbt-20240429-182027_1/base.yaml",
"checkpoint": "/workspace/cope2n-ai-fi/weights/models/sdsvkvu/key_value_understanding_for_sbt-20240416-153954/checkpoints/best_model.pth" "checkpoint": "/workspace/cope2n-ai-fi/weights/models/sdsvkvu/key_value_understanding_for_sbt-20240429-182027_1/checkpoints/best_model.pth"
} }
} }