how to run the code in single gpu?

[Pixie8888]

Hi,
Thanks for releasing the code. I want to run your code in single gpu. I am trying running this command:
"python run.py --method COS --name COS --epochs 30 --lr 0.01 --batch_size 24",
but it gives this error:

Could you please give some suggestion on how to run the code in the single gpu? Thanks!

[fcdl94]

Hello, please check the README.md file.
Moreover, to run experiments, I suggest using the files in the run folder.

In your specific case, you should use the following:
python -m torch.distributed.launch --master_port ${port} --nproc_per_node=1 run.py --num_workers 4 ... your parameters here ...

[Pixie8888]

thanks.