name: ImageNet

conda_env: conda.yaml

entry_points:

  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}"
