From 704a6a5dc9f4aedcced9efb82f72cd453c3d4b43 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Thu, 21 Nov 2024 23:28:34 +0100
Subject: [PATCH] fix: pretrained_instance properly mapped into
--pretrained-model-instance
---
src/Taggers.jl | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/Taggers.jl b/src/Taggers.jl
index 4844907..b66ef40 100644
--- a/src/Taggers.jl
+++ b/src/Taggers.jl
@@ -236,7 +236,9 @@ function run(tagger, switch, kwargs)
Base.run(Cmd(`$cmd $switch $args`; dir=tagger.backend_dir))
end
-function train(tagger::Tagger; kwargs...)
+function train(tagger::Tagger; pretrained_instance=None, kwargs...)
+ kwargs = Dict{Symbol, Any}(kwargs)
+ kwargs[:pretrained_model_instance] = pretrained_instance
ret = run(tagger, "train", kwargs)
@assert isdir(modeldir(tagger))
return ret
--
GitLab