From b65d5a14bc56a5fa8a63353284750c7b68576e55 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Sat, 20 May 2023 12:45:05 +0200
Subject: [PATCH] pretrained_model_instance passed to make_dataset
---
Manifest.toml | 2 +-
src/taggingbackends/main.py | 2 ++
2 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/Manifest.toml b/Manifest.toml
index 4f9b94d..2639199 100644
--- a/Manifest.toml
+++ b/Manifest.toml
@@ -310,7 +310,7 @@ version = "1.8.0"
[[deps.PlanarLarvae]]
deps = ["DelimitedFiles", "HDF5", "JSON3", "LinearAlgebra", "MAT", "Meshes", "OrderedCollections", "Random", "SHA", "StaticArrays", "Statistics", "StatsBase", "StructTypes"]
-git-tree-sha1 = "af2845a57442b046685d2b230ac30da50bb8314d"
+git-tree-sha1 = "bff9bdff96050dffcb681e950ac2547e33c53417"
repo-rev = "dev"
repo-url = "https://gitlab.pasteur.fr/nyx/planarlarvae.jl"
uuid = "c2615984-ef14-4d40-b148-916c85b43307"
diff --git a/src/taggingbackends/main.py b/src/taggingbackends/main.py
index e59ce4d..931c471 100644
--- a/src/taggingbackends/main.py
+++ b/src/taggingbackends/main.py
@@ -190,6 +190,8 @@ def main(fun=None):
elif reuse_h5files:
# deprecated
logging.info("option --reuse-h5files is ignored in the absence of --trxmat-only")
+ if pretrained_model_instance is not None:
+ make_dataset_kwargs["pretrained_model_instance"] = pretrained_model_instance
if include_all:
make_dataset_kwargs["include_all"] = include_all
if seed is not None:
--
GitLab