Skip to content

Commit

Permalink
fix: configuration of dspn runs
Browse files Browse the repository at this point in the history
  • Loading branch information
Francesco Stablum committed Nov 8, 2021
1 parent 10a5f30 commit a0b83e0
Show file tree
Hide file tree
Showing 7 changed files with 55 additions and 3 deletions.
14 changes: 14 additions & 0 deletions launch_run.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
#!/bin/bash

MODEL=$1
CONF=$2

if [ -z "$MODEL" ]; then
echo "argument 1 (model) was not specified"
fi

if [ -z "$CONF" ]; then
echo "argument 2 (conf) was not specified"
fi

screen -S $CONF -d -m python3 models/${MODEL}.py $CONF
13 changes: 13 additions & 0 deletions model_config/dspn_budget_normal.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
experiment_name: dspn_first
layers_width: 256
latent_dim: 32
activation_function: "ELU"
depth: 2
weight_decay: 1.0e-4
max_epochs: 1000
rel_name: 'budget'
batch_size: 1024
divide_output_layer: True
latent_l1_norm: 10.0
max_set_size: 200
dspn_iter: 4
13 changes: 13 additions & 0 deletions model_config/dspn_result_normal.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
experiment_name: dspn_first
layers_width: 256
latent_dim: 32
activation_function: "ELU"
depth: 2
weight_decay: 1.0e-4
max_epochs: 20
rel_name: 'result'
batch_size: 1024
divide_output_layer: True
latent_l1_norm: 10.0
max_set_size: 200
dspn_iter: 4
13 changes: 13 additions & 0 deletions model_config/dspn_result_short.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
experiment_name: dspn_first
layers_width: 3
latent_dim: 2
activation_function: "ELU"
depth: 5
weight_decay: 0
max_epochs: 2
rel_name: 'result'
batch_size: 1024
divide_output_layer: True
latent_l1_norm: 10.0
max_set_size: 200
dspn_iter: 4
2 changes: 1 addition & 1 deletion model_config/result_normal_bottleneck.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ latent_dim: 32
activation_function: "ELU"
depth: 2
weight_decay: 1.0e-4
max_epochs: 1000
max_epochs: 20
rel_name: 'result'
batch_size: 1024
divide_output_layer: True
Expand Down
1 change: 0 additions & 1 deletion model_config/result_short_test.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -9,4 +9,3 @@ rel_name: 'result'
batch_size: 1024
divide_output_layer: True
latent_l1_norm: 10.0
max_set_size: 200
2 changes: 1 addition & 1 deletion models/dspn_autoencoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ def __init__(self, **kwargs):
kwargs['item_dim'],
self.max_set_size,
kwargs['item_dim'],
10, # number of iteration on the decoder # FIXME: to conf
kwargs['dspn_iter'], # number of iteration on the decoder # FIXME: to conf
800, # inner learning rate # FIXME: to conf
)

Expand Down

0 comments on commit a0b83e0

Please sign in to comment.