Commit 235146c5 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added hyperparameter logging for run trial and cluster number

parent 2ec3c08c
Pipeline #88838 failed with stage
in 120 minutes and 30 seconds
......@@ -37,11 +37,23 @@ rule explore_encoding_dimension_and_loss_function:
"dimension_and_loss_experiments/trained_weights/GMVAE_loss={loss}_encoding={encs}_run_1_final_weights.h5",
),
shell:
"pipenv run python -m deepof.train_model --train-path {input.data_path} --val-num 10 --components 10 \
--input-type coords --predictor 0 --variational True --loss {wildcards.loss} --kl-warmup 20 --mmd-warmup 20 \
--encoding-size {wildcards.encs} --batch-size 256 --window-size 11 --window-step 11 --exclude-bodyparts \
Tail_base,Tail_1,Tail_2,Tail_tip,Spine_2 --stability-check 3 --logparam encoding --output-path \
{outpath}dimension_and_loss_experiments"
"pipenv run python -m deepof.train_model "
"--train-path {input.data_path} "
"--val-num 10 "
"--components 10 "
"--input-type coords "
"--predictor 0 "
"--variational True "
"--loss {wildcards.loss} "
"--kl-warmup 20 "
"--mmd-warmup 20 "
"--encoding-size {wildcards.encs} "
"--batch-size 256 "
"--window-size 11 "
"--window-step 11 "
"--exclude-bodyparts Tail_base,Tail_1,Tail_2,Tail_tip,Spine_2 "
"--stability-check 3 "
"--output-path {outpath}dimension_and_loss_experiments"
# rule explore_number_of_components:
......
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