- Configure environment and download example dataset.
- Copy
config.py.template
toconfig.py
and fill in missing entries to reflect your system. - Do the same for
db/credentials.py
- run
python setup.py install
- Download the data files from http://bit.ly/bsds_tfrecords. Place the files in the
self.tf_records
directory that you specified inconfig.py
.
- You can ignore errors with psql install and database creation.
- Copy
- Train a model.
CUDA_VISIBLE_DEVICES=0 python run_job.py --experiment=BSDS500_combos_100_no_aux --no_db --model=BSDS_vgg_cheap_deepest_final_simple --train=BSDS500_100_jk --val=BSDS500_100_jk
- Test a model.
CUDA_VISIBLE_DEVICES=0 python run_job.py --experiment=BSDS500_test --model=BSDS_vgg_cheap_deepest_final_simple --no_db --ckpt=<path_to_model_checkpoint> --placeholders --test --out_dir=bsds_landscape --train=BSDS500_test_landscape --val=BSDS500_test_landscape
CUDA_VISIBLE_DEVICES=0 python run_job.py --experiment=BSDS500_test --model=BSDS_vgg_cheap_deepest_final_simple --no_db --ckpt=<path_to_model_checkpoint> --placeholders --test --out_dir=bsds_portrait --train=BSDS500_test_portrait --val=BSDS500_test_portrait
- You will need to download the BSDS500 test set, and set the paths in
datasets/BSDS500_test_landscape.py
appropriately.
-
Configure environment and DB.
- Copy
config.py.template
toconfig.py
and fill in missing entries to reflect your system. - Do the same for
db/credentials.py
- run
python setup.py install
- Copy
-
Create an experiment.
- See
experiments/nist_baseline.py
for an example experiment definition. - See
models/seung_unet.py
for an example model specification (included in the experiment definition). - See
datasets/cluttered_nist_baseline.py
for an example dataset class (included in the experiment definition). - Initialize the DB and load an experiment:
python build_experiments.py --experiment=nist_baseline --initialize
- I manually access the db with
psql cluttered_nist -h 127.0.0.1 -d cluttered_nist
.
- I manually access the db with
- See
-
Run an experiment.
- A single job from the DB:
CUDA_VISIBLE_DEVICES=0 python run_job.py
- A single job without the DB:
CUDA_VISIBLE_DEVICES=0 python run_job.py --no_db --experiment=nist_baseline --model=seung_unet --train=cluttered_nist_baseline --test=cluttered_nist_baseline
- A local worker that continues until the DB is exhausted:
bash start_worker.sh
- Fill the p-nodes with workers running in Dockers:
bash docker_workers.sh
- A single job from the DB:
-
Manually access the DB - psql cluttered_nist -h 127.0.0.1 -d cluttered_nist
-
Run and kill docker jobs - Run docker job
bash docker_workers.sh
- Kill docker jobspython utils/docker_kill.py bash
- Get docker pidsdocker ps
- Get docker job stdout `docker logs