name: PascalVOC2012

conda_env: conda.yaml

entry_points:

  download:
    parameters:
      output_path: path
    command: "export PYTHONPATH=$PWD/../../code:$PYTHONPATH && python ../../code/scripts/download_dataset.py {output_path}"

  main:
    parameters:
      config_path: path
      num_gpus: float
    command: "sh ../setup_apex.sh && export PYTHONPATH=$PWD/../../code:$PYTHONPATH && python -m torch.distributed.launch --nproc_per_node={num_gpus} --use_env -m py_config_runner ../../code/scripts/training.py {config_path}"
