stylegan2-pytorch icon indicating copy to clipboard operation
stylegan2-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon rosinality

Train script issue

Open polimorfo opened this issue 3 years ago • 2 comments

When training the model in colab: %run -m torch.distributed.launch --nproc_per_node=1 --master_port=8890 train.py --batch 4 LMDB_PATH the following error raises: UsageError: option --nproc_per_node not recognized ( allowed: "nidtN:b:pD:l:rs:T:em:G" ) Does anybody know why ?

polimorfo avatar Sep 29 '21 11:09 polimorfo

I don't know much about colab, but it seems like that %run -m torch.distributed.launch does not corresponds to python -m torch.distributed.launch. Maybe this is the related issues. https://github.com/ipython/ipython/issues/8437

rosinality avatar Oct 03 '21 07:10 rosinality

Just changed the '%run' command for '!python' one. Also the module torch.distributed.launch didn´t seem to work and was migrated to torch.distributed.run. Model is training now :) Thanks for the advice.

polimorfo avatar Oct 07 '21 14:10 polimorfo