rasya-ramesh

DataParallel - AssertionError: Invalid device id

2

Traceback (most recent call last): File "upsnet/upsnet_end2end_train.py", line 407, in upsnet_train() File "upsnet/upsnet_end2end_train.py", line 176, in upsnet_train train_model = DataParallel(train_model, device_ids=[int(_) for _ in config.gpus.split(',')]).to(gpus[0]) File "/usr/local/lib/python3.6/dist-packages/torch/nn/parallel/data_parallel.py", line 133, in...