From 8b1a955b8aeaf226aa35a346b29cc7341391c0af Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Mon, 7 Dec 2020 22:30:52 +0100
Subject: [PATCH] Added CustomStopper class to train_utils.py, to start early
stopping only after annealing is over
---
deepof/train_model.py | 4 ++--
deepof_experiments.smk | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/deepof/train_model.py b/deepof/train_model.py
index 7d0efdb3..4975c33e 100644
--- a/deepof/train_model.py
+++ b/deepof/train_model.py
@@ -218,7 +218,7 @@ arena_dims = args.arena_dims
batch_size = args.batch_size
hypertun_trials = args.hpt_trials
encoding_size = args.encoding_size
-exclude_bodyparts = list(args.exclude_bodyparts.split(","))
+exclude_bodyparts = tuple(args.exclude_bodyparts.split(","))
gaussian_filter = args.gaussian_filter
hparams = args.hyperparameters
input_type = args.input_type
@@ -274,7 +274,7 @@ project_coords = project(
animal_ids=tuple([animal_id]),
arena="circular",
arena_dims=tuple([arena_dims]),
- exclude_bodyparts=(None if exclude_bodyparts == [""] else exclude_bodyparts),
+ exclude_bodyparts=exclude_bodyparts,
exp_conditions=treatment_dict,
path=train_path,
smooth_alpha=smooth_alpha,
diff --git a/deepof_experiments.smk b/deepof_experiments.smk
index 758fa3af..80c5a744 100644
--- a/deepof_experiments.smk
+++ b/deepof_experiments.smk
@@ -67,7 +67,7 @@ rule explore_encoding_dimension_and_loss_function:
shell:
"pipenv run python -m deepof.train_model "
"--train-path {input.data_path} "
- "--val-num 25 "
+ "--val-num 5 "
"--components {wildcards.k} "
"--input-type coords "
"--predictor 0 "
--
GitLab