Skip to content
Snippets Groups Projects
Commit a780ff0a authored by Erick Lavoie's avatar Erick Lavoie
Browse files

Fixed missing option in rm-edges scripts for CIFAR10

parent 487fd44c
No related branches found
No related tags found
No related merge requests found
......@@ -14,6 +14,7 @@ export PYTHONPATH=$TOOLS
# used by the simulator. For a list of available options for each
# command, run 'export PYTHONPATH=.; <command> --help'.
for RM_EDGES in 1 5; do
for MOMENTUM in '--learning-momentum 0.9' '--learning-momentum 0.0'; do
for C_GRAD in '--clique-gradient' ''; do
setup/meta.py \
--script $SCRIPT_DIR/`basename "$0"` \
......@@ -31,11 +32,12 @@ setup/nodes/google-fl.py \
setup/topology/d_cliques/greedy_swap.py \
--interclique fully-connected \
--max-clique-size 10 \
--max-steps 1000 |
--max-steps 1000 \
--remove-clique-edges $RM_EDGES |
setup/model/gn_lenet.py |
simulate/algorithm/d_sgd.py \
--batch-size 20 \
--learning-momentum 0.9 \
$MOMENTUM \
$C_GRAD \
--learning-rate 0.002 |
simulate/logger.py \
......@@ -44,4 +46,5 @@ simulate/logger.py \
simulate/run.py \
--nb-epochs 100;
done;
done;
done
......@@ -14,6 +14,7 @@ export PYTHONPATH=$TOOLS
# used by the simulator. For a list of available options for each
# command, run 'export PYTHONPATH=.; <command> --help'.
for RM_EDGES in 1 5; do
for MOMENTUM in '--learning-momentum 0.9' '--learning-momentum 0.0'; do
for C_GRAD in '--clique-gradient' ''; do
setup/meta.py \
--script $SCRIPT_DIR/`basename "$0"` \
......@@ -29,11 +30,12 @@ setup/nodes.py \
--nodes-per-class 10 10 10 10 10 10 10 10 10 10 \
--local-classes 1 |
setup/topology/d_cliques/ideal.py \
--remove-clique-edges $RM_EDGES \
--interclique fully-connected |
setup/model/gn_lenet.py |
simulate/algorithm/d_sgd.py \
--batch-size 20 \
--learning-momentum 0.9 \
$MOMENTUM \
$C_GRAD \
--learning-rate 0.002 |
simulate/logger.py \
......@@ -42,4 +44,5 @@ simulate/logger.py \
simulate/run.py \
--nb-epochs 100;
done;
done;
done
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