Commit 8b1a955b authored by lucas_miranda's avatar lucas_miranda
Browse files

Added CustomStopper class to train_utils.py, to start early stopping only after annealing is over

parent a0dede93
Pipeline #89054 failed with stage
in 27 minutes and 45 seconds
......@@ -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,
......
......@@ -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 "
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment