Skip to content
Snippets Groups Projects
Commit b65d5a14 authored by François  LAURENT's avatar François LAURENT
Browse files

pretrained_model_instance passed to make_dataset

parent 0de5752c
No related branches found
No related tags found
No related merge requests found
Pipeline #104087 passed
...@@ -310,7 +310,7 @@ version = "1.8.0" ...@@ -310,7 +310,7 @@ version = "1.8.0"
[[deps.PlanarLarvae]] [[deps.PlanarLarvae]]
deps = ["DelimitedFiles", "HDF5", "JSON3", "LinearAlgebra", "MAT", "Meshes", "OrderedCollections", "Random", "SHA", "StaticArrays", "Statistics", "StatsBase", "StructTypes"] 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-rev = "dev"
repo-url = "https://gitlab.pasteur.fr/nyx/planarlarvae.jl" repo-url = "https://gitlab.pasteur.fr/nyx/planarlarvae.jl"
uuid = "c2615984-ef14-4d40-b148-916c85b43307" uuid = "c2615984-ef14-4d40-b148-916c85b43307"
......
...@@ -190,6 +190,8 @@ def main(fun=None): ...@@ -190,6 +190,8 @@ def main(fun=None):
elif reuse_h5files: elif reuse_h5files:
# deprecated # deprecated
logging.info("option --reuse-h5files is ignored in the absence of --trxmat-only") 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: if include_all:
make_dataset_kwargs["include_all"] = include_all make_dataset_kwargs["include_all"] = include_all
if seed is not None: if seed is not None:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment