I am referring a git repository which uses following bash file for starting the training of a neural network. I want to understand the use of '--use_env' flag.
#! /bin/bash
CUDA_VISIBLE_DEVICES=2 python -m torch.distributed.launch --nproc_per_node=1 --master_port 2351 --use_env train.py --dataroot SOME_PATH_TO_DATA
In the above command, what is the importance of '--use_env' flag? I have also checked that if I remove that flag, I get error saying "train.py: error: unrecognized arguments: -m torch.distributed.launch" as shown below.
train.py: error: unrecognized arguments: -m torch.distributed.launch
Can anyone brief me about this flag?
